Spaces:

nvidia
/

audio-flamingo-3

Running on A100

SreyanG-NVIDIA commited on 11 days ago

Commit

377af38

verified ·

1 Parent(s): e110d57

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ non_lora_trainables = torch.load(
 non_lora_trainables = {
         (k[6:] if k.startswith("model.") else k): v for k, v in non_lora_trainables.items()
     }
-model.load_state_dict(non_lora_trainables, strict=False)
 model_think = PeftModel.from_pretrained(
         model_single,
         MODEL_BASE_THINK,

 non_lora_trainables = {
         (k[6:] if k.startswith("model.") else k): v for k, v in non_lora_trainables.items()
     }
+model_single.load_state_dict(non_lora_trainables, strict=False)
 model_think = PeftModel.from_pretrained(
         model_single,
         MODEL_BASE_THINK,