requirement

Stable3DGen/hi3dgen/pipelines/hi3dgen.py

@@ -92,21 +92,11 @@ class Hi3DGenPipeline(Pipeline):
         """
         Initialize the image conditioning model.
         """
-        dinov2_model = torch.hub.load(
-        repo_or_dir="/share/project/cwm/houyuan.chen/Stable3DGen/dinov2",
-        model=name,
-        source='local',         # <--- 这是最关键的修改
-        pretrained=False
 
-        )
-        dinov2_weight = torch.load("/share/project/cwm/houyuan.chen/Stable3DGen/dinov2_vitl14_reg4_pretrain.pth")
-        dinov2_model.load_state_dict(dinov2_weight)
-
-
-        # try:
-        #     dinov2_model = torch.hub.load(os.path.join(torch.hub.get_dir(), 'facebookresearch_dinov2_main'), name, source='local',pretrained=True)
-        # except:
-        #     dinov2_model = torch.hub.load('facebookresearch/dinov2', name, pretrained=True)
+        try:
+            dinov2_model = torch.hub.load(os.path.join(torch.hub.get_dir(), 'facebookresearch_dinov2_main'), name, source='local',pretrained=True)
+        except:
+            dinov2_model = torch.hub.load('facebookresearch/dinov2', name, pretrained=True)
         dinov2_model.eval()
         self.models['image_cond_model'] = dinov2_model
         transform = transforms.Compose([

requirements.txt

@@ -28,5 +28,4 @@ transformers==4.46.3
 trimesh==4.6.12
 wandb==0.20.1
 xformers==0.0.27.post2
-flash_attn==2.8.0.post2
 huggingface_hub==0.33.0