update

infer_api.py

@@ -360,13 +360,6 @@ class InferAPI:
         self.slrm_configs = slrm_configs
         self.refine_configs = refine_configs
 
-        repo_id = "hyz317/StdGEN"
-        all_files = list_repo_files(repo_id, revision="main")
-        for file in all_files:
-            if os.path.exists(file):
-                continue
-            hf_hub_download(repo_id, file, local_dir="./ckpt")
-
         # self.canonical_infer = InferCanonicalAPI(self.canonical_configs)
         # self.multiview_infer = InferMultiviewAPI(self.multiview_configs)
         # self.slrm_infer = InferSlrmAPI(self.slrm_configs)
@@ -810,6 +803,12 @@ class InferMultiviewAPI:
             self.results[k] = results[k]
         return results
 
+repo_id = "hyz317/StdGEN"
+all_files = list_repo_files(repo_id, revision="main")
+for file in all_files:
+    if os.path.exists(file):
+        continue
+    hf_hub_download(repo_id, file, local_dir="./ckpt")
 
 infer_canonicalize_config = {
     'config_path': './configs/canonicalization-infer.yaml',