Spaces:
Running
Running
disable cache for better compability
Browse files
app.py
CHANGED
|
@@ -108,6 +108,7 @@ def load_pipeline(model_name):
|
|
| 108 |
trust_remote_code=True,
|
| 109 |
torch_dtype=dtype,
|
| 110 |
device_map="auto",
|
|
|
|
| 111 |
token=access_token)
|
| 112 |
PIPELINES[model_name] = pipe
|
| 113 |
return pipe
|
|
|
|
| 108 |
trust_remote_code=True,
|
| 109 |
torch_dtype=dtype,
|
| 110 |
device_map="auto",
|
| 111 |
+
use_cache=False, # ← disable past-key-value caching
|
| 112 |
token=access_token)
|
| 113 |
PIPELINES[model_name] = pipe
|
| 114 |
return pipe
|