Spaces:
Paused
Paused
Update app.py
Browse files
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 |
)
|