Spaces:
Sleeping
Sleeping
stevessschen
commited on
Commit
•
e0416a0
1
Parent(s):
fa89470
Update app.py
Browse files
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 =
|
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
|