Spaces:
Running
on
Zero
Running
on
Zero
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
|