space

app.py

@@ -108,11 +108,11 @@ MAX_IMAGE_SIZE = 1024  # Currently not used.
 # Load the main diffusion model.
 repo_id = "QHL067/CrossFlow"
 filename = "pretrained_models/t2i_512px_clip_dimr.pth"
-checkpoint_path = hf_hub_download(repo_id=repo_id, filename=filename)
+# checkpoint_path = hf_hub_download(repo_id=repo_id, filename=filename)
 nnet = utils.get_nnet(**config.nnet)
 nnet = nnet.to(device)
-state_dict = torch.load(checkpoint_path, map_location=device)
-nnet.load_state_dict(state_dict)
+# state_dict = torch.load(checkpoint_path, map_location=device)
+# nnet.load_state_dict(state_dict)
 nnet.eval()
 
 # Initialize text model.