Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -12,7 +12,7 @@ def load_model():
|
|
| 12 |
"Reham1/mistral-diabetes-lora1",
|
| 13 |
torch_dtype="auto",
|
| 14 |
device_map="auto",
|
| 15 |
-
offload_folder="/tmp/offload",
|
| 16 |
token=API_TOKEN
|
| 17 |
)
|
| 18 |
generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
|
|
|
| 12 |
"Reham1/mistral-diabetes-lora1",
|
| 13 |
torch_dtype="auto",
|
| 14 |
device_map="auto",
|
| 15 |
+
offload_folder="/tmp/offload",
|
| 16 |
token=API_TOKEN
|
| 17 |
)
|
| 18 |
generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|