Leadmatic
/

tinyChat

Model card Files Files and versions Community

kishb87 commited on Apr 23, 2023

Commit

5858e96

·

1 Parent(s): 716f1ef

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -49,4 +49,5 @@ peft_model = PeftModel.from_pretrained(base_model, peft_model_id)
 inputs = tokenizer("""[INSERT INSTRUCTION HERE]""", return_tensors="pt")
 outputs = peft_model.generate(**inputs, max_length=300, do_sample=True)
-print(tokenizer.batch_decode(outputs, skip_special_tokens=True))

 inputs = tokenizer("""[INSERT INSTRUCTION HERE]""", return_tensors="pt")
 outputs = peft_model.generate(**inputs, max_length=300, do_sample=True)
+print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
+```