CustomNet

Runtime error

cocktailpeanut commited on Apr 9, 2024

Commit

1c2ea32

1 Parent(s): dd6adc1

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ model_path='./customnet_v1.pt'
 #if not os.path.exists(model_path):
 #    os.system(f'wget https://huggingface.co/TencentARC/CustomNet/resolve/main/customnet_v1.pt?download=true -P .')
 ckpt = torch.load(model_path, map_location="cpu")
-model.load_state_dict(ckpt)
 del ckpt
 model = model.to(device)

 #if not os.path.exists(model_path):
 #    os.system(f'wget https://huggingface.co/TencentARC/CustomNet/resolve/main/customnet_v1.pt?download=true -P .')
 ckpt = torch.load(model_path, map_location="cpu")
+model.load_state_dict(ckpt, strict=False)
 del ckpt
 model = model.to(device)