Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ print(datetime.fromtimestamp(time.time()))
|
|
| 30 |
model_name = "TheBloke/TinyLlama-1.1B-Chat-GGUF" # Change if using a different version
|
| 31 |
#model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0" # Change if using a different version
|
| 32 |
#model_name = "microsoft/phi-1_5" # Change if using a different version
|
| 33 |
-
model = AutoModelForCausalLM.from_pretrained("
|
| 34 |
|
| 35 |
print("tokenizer")
|
| 36 |
print(datetime.fromtimestamp(time.time()))
|
|
|
|
| 30 |
model_name = "TheBloke/TinyLlama-1.1B-Chat-GGUF" # Change if using a different version
|
| 31 |
#model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0" # Change if using a different version
|
| 32 |
#model_name = "microsoft/phi-1_5" # Change if using a different version
|
| 33 |
+
model = AutoModelForCausalLM.from_pretrained("tinyllama-1.1b-chat-v1.0.Q4_K_M.gguf", model_type="llama")
|
| 34 |
|
| 35 |
print("tokenizer")
|
| 36 |
print(datetime.fromtimestamp(time.time()))
|