Spaces:

QHL067
/

CrossFlow

Running on Zero

QHL067 commited on Mar 15

Commit

0cfd333

1 Parent(s): 485bed7

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ nnet_1.eval()
 filename = "pretrained_models/t2i_512px_clip_dimr.pth"
 checkpoint_path = hf_hub_download(repo_id=repo_id, filename=filename)
 nnet_2 = utils.get_nnet(**config_2.nnet)
-nnet_2 = nnet.to(device)
 state_dict = torch.load(checkpoint_path, map_location=device)
 nnet_2.load_state_dict(state_dict)
 nnet_2.eval()

 filename = "pretrained_models/t2i_512px_clip_dimr.pth"
 checkpoint_path = hf_hub_download(repo_id=repo_id, filename=filename)
 nnet_2 = utils.get_nnet(**config_2.nnet)
+nnet_2 = nnet_2.to(device)
 state_dict = torch.load(checkpoint_path, map_location=device)
 nnet_2.load_state_dict(state_dict)
 nnet_2.eval()