Luigi commited on
Commit
ea7c44f
·
1 Parent(s): 493107e

disable cache for better compability

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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