Fix ModuleNotFoundError by adding defake to sys.path

defake/blipmodels/blip.py

@@ -22,7 +22,7 @@ from timm.models.hub import download_cached_file
 
 class BLIP_Base(nn.Module):
     def __init__(self,                 
-                 med_config = '/home/sha/stable-diffusion/blipmodels/blipconfig/med_config.json',  
+                 med_config = 'blipconfig/med_config.json',  
                  image_size = 224,
                  vit = 'base',
                  vit_grad_ckpt = False,
@@ -36,6 +36,19 @@ class BLIP_Base(nn.Module):
         """               
         super().__init__()
         
+        # 1. 获取 blip.py 所在的绝对路径
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        
+        # 2. 拼接出正确的 json 路径: blipmodels/blipconfig/med_config.json
+        # 注意这里用的是 'blipconfig'，对应你截图里的文件夹名
+        med_config = os.path.join(current_dir, 'blipconfig', 'med_config.json')
+        
+        print(f"DEBUG: Loading med_config from: {med_config}")
+        
+        # 3. 检查一下文件是否存在，不存在就打印个警告
+        if not os.path.exists(med_config):
+            print(f"ERROR: Config file NOT found at {med_config}")
+        
         self.visual_encoder, vision_width = create_vit(vit,image_size, vit_grad_ckpt, vit_ckpt_layer)
         self.tokenizer = init_tokenizer()   
         med_config = BertConfig.from_json_file(med_config)
@@ -77,7 +90,7 @@ class BLIP_Base(nn.Module):
         
 class BLIP_Decoder(nn.Module):
     def __init__(self,                 
-                 med_config = '/home/sha/stable-diffusion/blipmodels/blipconfig/med_config.json',  
+                 med_config = 'blipconfig/med_config.json',  
                  image_size = 384,
                  vit = 'base',
                  vit_grad_ckpt = False,
@@ -92,6 +105,19 @@ class BLIP_Decoder(nn.Module):
         """            
         super().__init__()
         
+        # 1. 获取 blip.py 所在的绝对路径
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        
+        # 2. 拼接出正确的 json 路径: blipmodels/blipconfig/med_config.json
+        # 注意这里用的是 'blipconfig'，对应你截图里的文件夹名
+        med_config = os.path.join(current_dir, 'blipconfig', 'med_config.json')
+        
+        print(f"DEBUG: Loading med_config from: {med_config}")
+        
+        # 3. 检查一下文件是否存在，不存在就打印个警告
+        if not os.path.exists(med_config):
+            print(f"ERROR: Config file NOT found at {med_config}")
+        
         self.visual_encoder, vision_width = create_vit(vit,image_size, vit_grad_ckpt, vit_ckpt_layer)
         self.tokenizer = init_tokenizer()   
         med_config = BertConfig.from_json_file(med_config)