Views
No views yet
from transformers import AutoModel model = AutoModel.from_pretrained("tontiru/finetuning1") outputs = model.generate( tokenized_input, attention_mask=attention_mask, max_new_tokens=100, do_sample=False, repetition_penalty=1.2, pad_token_id=tokenizer.eos_token_id )[0]