Spaces:

adnlp
/

MulTiCast

Running

adnlp commited on Sep 6, 2025

Commit

295706c

verified ·

1 Parent(s): 9f2785e

Upload MultiModalTimer.py

Files changed (1) hide show

MultiModalTimer.py CHANGED Viewed

@@ -71,7 +71,7 @@ class MultiModalTimerModel(PreTrainedModel):
             text_model = AutoModelForCausalLM.from_pretrained(
                 "Qwen/Qwen2-1.5B-Instruct",
                 torch_dtype=torch.bfloat16,
-                device_map="auto",
                 attn_implementation="sdpa"
             ).model
             state_dict = text_model.state_dict()

             text_model = AutoModelForCausalLM.from_pretrained(
                 "Qwen/Qwen2-1.5B-Instruct",
                 torch_dtype=torch.bfloat16,
+                device_map="cpu",
                 attn_implementation="sdpa"
             ).model
             state_dict = text_model.state_dict()