multimodal-deepfakes

Runtime error

Kimata commited on Jul 3, 2023

Commit

11dd612

1 Parent(s): d7b6ade

load model stata dict correctly

Files changed (2) hide show

__pycache__/pipeline.cpython-39.pyc CHANGED Viewed

Binary files a/__pycache__/pipeline.cpython-39.pyc and b/__pycache__/pipeline.cpython-39.pyc differ

pipeline.py CHANGED Viewed

@@ -202,7 +202,7 @@ def load_audio_model():
     #Load pretrained ckpt.
     model_dict = model.state_dict()
     ckpt = torch.load('pre_trained_DF_RawNet2.pth', map_location=torch.device('cpu'))
-    model = model.load_state_dict(ckpt, model_dict)
     return model
 audio_label_map = {

     #Load pretrained ckpt.
     model_dict = model.state_dict()
     ckpt = torch.load('pre_trained_DF_RawNet2.pth', map_location=torch.device('cpu'))
+    model.load_state_dict(ckpt, model_dict)
     return model
 audio_label_map = {