Arnab Das commited on
Commit
ed8b10d
·
1 Parent(s): 2324266
Files changed (1) hide show
  1. manipulate_model/utils.py +1 -1
manipulate_model/utils.py CHANGED
@@ -17,7 +17,7 @@ def get_config_and_model(model_root="manipulate_model/demo-model/audio"):
17
  if isinstance(config.model.decoder, str):
18
  config.model.decoder = OmegaConf.load(config.model.decoder)
19
 
20
- model = Model.from_pretrained("arnabdas8901/manipulation_detection_transformer", config=config)
21
  #weights = torch.load(os.path.join(model_root, "weights.pt"))
22
  #model.load_state_dict(weights["model_state_dict"])
23
 
 
17
  if isinstance(config.model.decoder, str):
18
  config.model.decoder = OmegaConf.load(config.model.decoder)
19
 
20
+ model = Model._from_pretrained("arnabdas8901/manipulation_detection_transformer", config=config)
21
  #weights = torch.load(os.path.join(model_root, "weights.pt"))
22
  #model.load_state_dict(weights["model_state_dict"])
23