macadeliccc
/

WestLake-7B-v2-laser-truthy-dpo

Text Generation

text-generation-inference

Model card Files Files and versions

macadeliccc commited on Jan 30, 2024

Commit

67ff7a2

·

verified ·

1 Parent(s): 62c4197

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
-text = "Come up with an idea for a TV show and write a short pilot script"
 inputs = tokenizer(text, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=4096)

 model = AutoModelForCausalLM.from_pretrained(model_id)
+text = "Create an idea for a TV show and write a short pilot script"
 inputs = tokenizer(text, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=4096)