Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModel
|
|
5 |
file = 'llama-2-7b.Q4_K_M.gguf'
|
6 |
NO_GPU = 0
|
7 |
GPU_LAYERS = 50
|
8 |
-
model = AutoModel.from_pretrained("TheBloke/Llama-2-7B-GGUF
|
9 |
model_file=file, model_type="llama", gpu_layers=NO_GPU)
|
10 |
prompt = "AI is going to"
|
11 |
|
|
|
5 |
file = 'llama-2-7b.Q4_K_M.gguf'
|
6 |
NO_GPU = 0
|
7 |
GPU_LAYERS = 50
|
8 |
+
model = AutoModel.from_pretrained("TheBloke/Llama-2-7B-GGUF" + file,
|
9 |
model_file=file, model_type="llama", gpu_layers=NO_GPU)
|
10 |
prompt = "AI is going to"
|
11 |
|