StreamingT2V

Runtime error

hpoghos commited on Apr 5

Commit

6be531d

•

1 Parent(s): 373be50

Update t2v_enhanced/model_init.py

Files changed (1) hide show

t2v_enhanced/model_init.py CHANGED Viewed

@@ -101,7 +101,7 @@ def init_streamingt2v_model(ckpt_file, result_fol):
         model = cli.model
         model.load_state_dict(torch.load(
-            cli.config["ckpt"].as_posix())["state_dict"])
     return cli, model

         model = cli.model
         model.load_state_dict(torch.load(
+            cli.config["ckpt"].as_posix(), map_location=torch.device('cpu'))["state_dict"], strict=False)
     return cli, model