Update app.py
Browse files
app.py
CHANGED
|
@@ -5,9 +5,11 @@ from huggingface_hub import login
|
|
| 5 |
import os
|
| 6 |
|
| 7 |
file = 'llama-2-7b.Q5_0.gguf'
|
|
|
|
|
|
|
| 8 |
NO_GPU = 0
|
| 9 |
GPU_LAYERS = 50
|
| 10 |
-
model = AutoModel.from_pretrained("
|
| 11 |
model_file=file, model_type="llama", gpu_layers=NO_GPU)
|
| 12 |
|
| 13 |
access_token = os.getenv('HF_TOKEN2')
|
|
|
|
| 5 |
import os
|
| 6 |
|
| 7 |
file = 'llama-2-7b.Q5_0.gguf'
|
| 8 |
+
|
| 9 |
+
|
| 10 |
NO_GPU = 0
|
| 11 |
GPU_LAYERS = 50
|
| 12 |
+
model = AutoModel.from_pretrained("valencar/llamm",
|
| 13 |
model_file=file, model_type="llama", gpu_layers=NO_GPU)
|
| 14 |
|
| 15 |
access_token = os.getenv('HF_TOKEN2')
|