Spaces:

Plachta
/

VALL-E-X

Runtime error

Plachta commited on Apr 20

Commit

1e529fc

verified ·

1 Parent(s): e448a2c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ model = VALLE(
         prepend_bos=True,
         num_quantizers=NUM_QUANTIZERS,
     ).to(device)
-checkpoint = torch.load("./epoch-10.pt", map_location='cpu')
 missing_keys, unexpected_keys = model.load_state_dict(
     checkpoint["model"], strict=True
 )

         prepend_bos=True,
         num_quantizers=NUM_QUANTIZERS,
     ).to(device)
+checkpoint = torch.load("./epoch-10.pt", map_location='cpu', weights_only=True)
 missing_keys, unexpected_keys = model.load_state_dict(
     checkpoint["model"], strict=True
 )