akoksal commited on
Commit
e500d6a
1 Parent(s): ad00c08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -14,8 +14,7 @@ from transformers import AutoTokenizer, pipeline, AutoModelForCausalLM
14
 
15
 
16
  tokenizer = AutoTokenizer.from_pretrained("akoksal/LongForm-OPT-2.7B")
17
- model = AutoModelForCausalLM.from_pretrained("akoksal/LongForm-OPT-2.7B", load_in_8bit=True, device_map='auto')
18
- generate = pipeline('text-generation', model=model, tokenizer=tokenizer)
19
 
20
 
21
  def predict(instruction, topp, max_length, temperature):
 
14
 
15
 
16
  tokenizer = AutoTokenizer.from_pretrained("akoksal/LongForm-OPT-2.7B")
17
+ generate = pipeline('text-generation', model="akoksal/LongForm-OPT-2.7B", tokenizer=tokenizer)
 
18
 
19
 
20
  def predict(instruction, topp, max_length, temperature):