update

asset3d_gen/models/delight_model.py

@@ -28,7 +28,7 @@ class DelightingModel(object):
         device: str = "cuda",
         seed: int = 0,
     ) -> None:
-        self.model_path = model_path
+        
         self.image_guide_scale = image_guide_scale
         self.text_guide_scale = text_guide_scale
         self.num_infer_step = num_infer_step
@@ -40,18 +40,19 @@ class DelightingModel(object):
         self.device = device
         self.pipeline = None # lazy load model adapt to @spaces.GPU
         
+        if model_path is None:
+            suffix = "hunyuan3d-delight-v2-0"
+            model_path = snapshot_download(
+                repo_id="tencent/Hunyuan3D-2", allow_patterns=f"{suffix}/*"
+            )
+            model_path = os.path.join(model_path, suffix)
+
+        self.model_path = model_path
+
     def _lazy_init_pipeline(self):
         if self.pipeline is None:
-            model_path = self.model_path
-            if model_path is None:
-                suffix = "hunyuan3d-delight-v2-0"
-                model_path = snapshot_download(
-                    repo_id="tencent/Hunyuan3D-2", allow_patterns=f"{suffix}/*"
-                )
-                model_path = os.path.join(model_path, suffix)
-
             pipeline = StableDiffusionInstructPix2PixPipeline.from_pretrained(
-                model_path,
+                self.model_path,
                 torch_dtype=torch.float16,
                 safety_checker=None,
             )

asset3d_gen/models/sr_model.py

@@ -73,8 +73,16 @@ class ImageRealESRGAN:
             sys.modules["torchvision.transforms.functional_tensor"] = functional_tensor
 
         self.outscale = outscale
-        self.model_path = model_path
         self.upsampler = None
+
+        if model_path is None:
+            suffix = "super_resolution"
+            model_path = snapshot_download(
+                repo_id="xinjjj/RoboAssetGen", allow_patterns=f"{suffix}/*"
+            )
+            model_path = os.path.join(model_path, suffix, "RealESRGAN_x4plus.pth")
+
+        self.model_path = model_path
         
     def _lazy_init(self):
         if self.upsampler is None:
@@ -91,17 +99,9 @@ class ImageRealESRGAN:
                 scale=4,
             )
 
-            model_path = self.model_path
-            if model_path is None:
-                suffix = "super_resolution"
-                model_path = snapshot_download(
-                    repo_id="xinjjj/RoboAssetGen", allow_patterns=f"{suffix}/*"
-                )
-                model_path = os.path.join(model_path, suffix, "RealESRGAN_x4plus.pth")
-
             self.upsampler = RealESRGANer(
                 scale=4,
-                model_path=model_path,
+                model_path=self.model_path,
                 model=model,
                 pre_pad=0,
                 half=True,