LenguajeNaturalAI
/

leniachat-gemma-2b-v0

Text Generation

text-generation-inference

multi-task-learning

instruction-tuning

Inference Endpoints

Model card Files Files and versions Community

avacaondata commited on Apr 13

Commit

e5691dc

•

1 Parent(s): 05cc9e7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -74,7 +74,7 @@ messages = [
 ]
 input_ids = tokenizer.apply_chat_template(messages, tokenize=True, return_tensors="pt")
 with torch.no_grad():
-  output = model.generate(**input_ids, max_length=50)
 print(tokenizer.decode(output[0], skip_special_tokens=True))
 ```

 ]
 input_ids = tokenizer.apply_chat_template(messages, tokenize=True, return_tensors="pt")
 with torch.no_grad():
+  output = model.generate(input_ids, max_new_tokens=50)
 print(tokenizer.decode(output[0], skip_special_tokens=True))
 ```