K

app.py

@@ -351,7 +351,8 @@ if settings.env.use_line_mask:
 
 settings.env.model_path = hf_hub_download(repo_id="hanquansanren/DvD", filename="model1852000.pt", token=token)
 # model.cpu().load_state_dict(dist_util.load_state_dict(settings.env.model_path, map_location="cpu"), strict=False)
-model.cpu().load(settings.env.model_path, map_location="cpu", strict=False)
+model_ckpt = torch.load(settings.env.model_path, map_location='cpu')
+model.cpu().load_state_dict(model_ckpt, strict=False)
 logger.log(f"Model loaded with {settings.env.model_path}")
 
 model.to('cuda')