mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-04 12:10:40 +08:00
update
This commit is contained in:
1 parent
c9a65db512
commit
05a6c509fb
2 files changed
+2
-2
No files matched your search
@@ -20,7 +20,7 @@ def loss(labels, logits):
|
||||
|
||||
class CTRLGenerator():
|
||||
|
||||
def __init__(self, control_code="Writing Text:", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0):
|
||||
def __init__(self, control_code="Writing Text: ", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0):
|
||||
|
||||
self.generate_num=generate_num
|
||||
model_dir = "generator/ctrl/training_utils/seqlen256_v1.ckpt/"
|
||||
|
||||
@@ -23,4 +23,4 @@ rooms:
|
||||
haunted_hospital: ["lobby", "hallway", "parking", "roof", "pharmacy"]
|
||||
|
||||
contexts:
|
||||
apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind."
|
||||
apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind. "
|
||||
Reference in new issue
Block a user