Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,6 @@ def load_model():
|
|
| 50 |
base_model_id,
|
| 51 |
device_map="auto",
|
| 52 |
torch_dtype=torch.float16,
|
| 53 |
-
use_cache=True,
|
| 54 |
trust_remote_code=True
|
| 55 |
)
|
| 56 |
|
|
|
|
| 50 |
base_model_id,
|
| 51 |
device_map="auto",
|
| 52 |
torch_dtype=torch.float16,
|
|
|
|
| 53 |
trust_remote_code=True
|
| 54 |
)
|
| 55 |
|