debug

app.py

@@ -144,7 +144,7 @@ class Handler:
         self.vae.to(self.device)
         self.vae.eval()
         self.preprocesser = load_preprocessor(self.device)
-        self.revq = ReVQ.from_pretrained(repo_id="AndyRaoTHU/revq-512T")
+        self.revq = ReVQ.from_pretrained("AndyRaoTHU/revq-512T")
         self.revq.to(self.device)
         self.revq.eval()
 

revq/models/revq.py

@@ -37,6 +37,7 @@ class Viewer:
 class ReVQ(PyTorchModelHubMixin, nn.Module):
     @classmethod
     def _from_pretrained(cls, repo_id: str, **kwargs):
+        print(f"Loading ReVQ model from {repo_id}...")
         config_path = hf_hub_download(repo_id=repo_id, filename="512T_NC=16384.yaml")
         ckpt_path = hf_hub_download(repo_id=repo_id, filename="ckpt.pth")