pablo-rf commited on
Commit
4756dc2
1 Parent(s): 9a094f3

Update How to Use

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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 brilla con forza no ceo, e "
57
 
58
  model_id = "proxectonos/FLOR-1.3B-GL"
59
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
60
  generator = pipeline(
61
  "text-generation",
62
- model=model_id,
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,