Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ MODEL_NAME = "tiiuae/falcon-7b-instruct"
|
|
51 |
|
52 |
llama_pipeline = pipeline(
|
53 |
"text-generation",
|
54 |
-
model=
|
55 |
torch_dtype=torch.float16,
|
56 |
device_map="auto",
|
57 |
)
|
|
|
51 |
|
52 |
llama_pipeline = pipeline(
|
53 |
"text-generation",
|
54 |
+
model=MODEL_NAME,
|
55 |
torch_dtype=torch.float16,
|
56 |
device_map="auto",
|
57 |
)
|