Raffaele Terribile commited on
Commit
534358c
·
unverified ·
1 Parent(s): ce72d8d

Ripristina il task originale

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ class FirstAgent:
46
  tokenizer = AutoTokenizer.from_pretrained(model_id)
47
  model = AutoModelForCausalLM.from_pretrained(model_id) # ~500MB
48
  model = pipeline(
49
- task="text-generation-inference",
50
  tokenizer=tokenizer,
51
  model=model
52
  )
 
46
  tokenizer = AutoTokenizer.from_pretrained(model_id)
47
  model = AutoModelForCausalLM.from_pretrained(model_id) # ~500MB
48
  model = pipeline(
49
+ task="text-generation",
50
  tokenizer=tokenizer,
51
  model=model
52
  )