Tomtom84 commited on
Commit
1bad3fb
·
verified ·
1 Parent(s): 4520cbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -58,7 +58,6 @@ async def load_models():
58
  model = AutoModelForCausalLM.from_pretrained(
59
  MODEL_REPO,
60
  low_cpu_mem_usage=True,
61
- return_legacy_cache=True,
62
  device_map={"": 0} if device == "cuda" else None,
63
  torch_dtype=torch.bfloat16 if device == "cuda" else None,
64
  )
 
58
  model = AutoModelForCausalLM.from_pretrained(
59
  MODEL_REPO,
60
  low_cpu_mem_usage=True,
 
61
  device_map={"": 0} if device == "cuda" else None,
62
  torch_dtype=torch.bfloat16 if device == "cuda" else None,
63
  )