Views
No views yet
pip install transformers1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_name = "Lineimos/French-lora-finetuned-Llama3-8B-novel-maupassant"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name)
7
8prompt = ("Voici une partie d'un roman. Veuillez vous en inspirer pour rédiger la suite:
9Il parlait en gaillard tranquille qui connaît la vie, et il souriait en regardant passer la foule.
10Mais tout d’un coup il se mit à tousser, et s’arrêta pour laisser finir la quinte, puis, d’un ton découragé :\n
11– Est-ce pas assommant de ne pouvoir se débarrasser de cette bronchite ?
12Et nous sommes en plein été. Oh ! cet hiver, j’irai me guérir à Menton. Tant pis, ma foi, la santé avant tout.")
13
14inputs = tokenizer(prompt, return_tensors="pt")
15outputs = model.generate(**inputs, max_new_tokens=100)
16print(tokenizer.decode(outputs[0], skip_special_tokens=True))



