medxiaorudan
/

CodeLlama_CPP_FineTuned

Model card Files Files and versions Community

medxiaorudan commited on Jan 24

Commit

064f9f6

•

1 Parent(s): 7f3dac0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,7 +62,7 @@ sequences = pipeline(
     temperature=0.1,
     top_p=0.95,
     num_return_sequences=1,
-    pad_token_id=tokenizer.eos_token_id,
     max_length=400,
     add_special_tokens=False
 )

     temperature=0.1,
     top_p=0.95,
     num_return_sequences=1,
+    eos_token_id=tokenizer.eos_token_id,
     max_length=400,
     add_special_tokens=False
 )