Update How to Use
Browse files
README.md
CHANGED
@@ -53,13 +53,14 @@ It can perform text-generation tasks and be fine-tuned for specific scenarios.
|
|
53 |
import torch
|
54 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
55 |
|
56 |
-
input_text = "Hoxe fai un bo día. O sol
|
57 |
|
58 |
model_id = "proxectonos/FLOR-1.3B-GL"
|
59 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
|
|
60 |
generator = pipeline(
|
61 |
"text-generation",
|
62 |
-
model=
|
63 |
tokenizer=tokenizer,
|
64 |
torch_dtype=torch.bfloat16,
|
65 |
trust_remote_code=True,
|
|
|
53 |
import torch
|
54 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
55 |
|
56 |
+
input_text = "Hoxe fai un bo día. O sol "
|
57 |
|
58 |
model_id = "proxectonos/FLOR-1.3B-GL"
|
59 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
60 |
+
model = AutoModelForCausalLM.from_pretrained(model_id)
|
61 |
generator = pipeline(
|
62 |
"text-generation",
|
63 |
+
model=model,
|
64 |
tokenizer=tokenizer,
|
65 |
torch_dtype=torch.bfloat16,
|
66 |
trust_remote_code=True,
|