Spaces:

10Aizen01
/

Engine-Builder-Beta

Sleeping

10Aizen01 commited on Apr 5

Commit

f03bf5c

verified ·

1 Parent(s): 8c50e84

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_id)
 base_model = AutoModelForCausalLM.from_pretrained(
     base_model_id,
     torch_dtype=torch.float32,
-    device_map={"": "cpu"}
     low_cpu_mem_usage=True
 )

 base_model = AutoModelForCausalLM.from_pretrained(
     base_model_id,
     torch_dtype=torch.float32,
+    device_map={"": "cpu"},
     low_cpu_mem_usage=True
 )