stevessschen commited on
Commit
e0416a0
1 Parent(s): fa89470

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoTokenizer
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
 
13
  tokenizer = AutoTokenizer.from_pretrained("stevessschen/llama-2-7b-miniguanaco")
14
- model = AutoModelForCausalLM.from_pretrained("stevessschen/llama-2-7b-miniguanaco")
15
 
16
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=200)
17
  # Run text generation pipeline with our next model
 
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
 
13
  tokenizer = AutoTokenizer.from_pretrained("stevessschen/llama-2-7b-miniguanaco")
14
+ model = TFAutoModelForCausalLM.from_pretrained("stevessschen/llama-2-7b-miniguanaco")
15
 
16
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=200)
17
  # Run text generation pipeline with our next model