support full-parameter PPO

This commit is contained in:
hiyouga committed 2023-11-16 02:08:04 +08:00
1 parent 8350bcf85d
commit ce78303600
20 files changed
+288 -145

No files matched your search

+1 -3
View File
@@ -66,6 +66,4 @@ class UltraChat(datasets.GeneratorBasedBuilder):
"from": "human" if i % 2 == 0 else "gpt",
"value": content[i]
} for i in range(len(content))]
yield key, {
"conversations": conversations
}
yield key, {"conversations": conversations}