Update modeling.py

modeling.py

@@ -4,7 +4,7 @@ from transformers import CLIPModel
 from peft import LoraConfig, get_peft_model
 
 class MLP(nn.Module):
-    def __init__(self, input_dim, hidden_dim1, hidden_dim2, output_dim,dropout_rate=0.5):
+    def __init__(self, input_dim=768, hidden_dim1=512, hidden_dim2=256, output_dim=8,dropout_rate=0.5):
         super(MLP, self).__init__()
         self.fc1 = nn.Linear(input_dim, hidden_dim1)
         self.relu1 = nn.ReLU()
@@ -24,7 +24,7 @@ class MLP(nn.Module):
         return x
     
 class clip_lora_model(nn.Module):
-    def __init__(self, input_dim, hidden_dim1, hidden_dim2, output_dim,dropout_rate=0.5,r=16,lora_alpha=8):
+    def __init__(self, input_dim=768, hidden_dim1=512, hidden_dim2=256, output_dim=8,dropout_rate=0.5,r=16,lora_alpha=8):
         super(clip_lora_model, self).__init__()
         self.output_dim=output_dim
         self.mlp = MLP(input_dim, hidden_dim1, hidden_dim2, output_dim,dropout_rate)
@@ -45,7 +45,7 @@ class clip_lora_model(nn.Module):
         )
         self.model = get_peft_model(encoder, config)
         
-    def forward(self, x,feat=None,flags=None):
+    def forward(self, x):
         model_outputs = self.model(x)
         image_embeds = model_outputs[1]
         model_outputs = self.proj(image_embeds)