Update modeling.py

modeling.py

@@ -24,7 +24,7 @@ class MLP(nn.Module):
         return x
     
 class clip_lora_model(nn.Module):
-    def __init__(self, input_dim=768, hidden_dim1=512, hidden_dim2=256, output_dim=8,dropout_rate=0.5,r=16,lora_alpha=8):
+    def __init__(self, input_dim=768, hidden_dim1=512, hidden_dim2=256, output_dim=1,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)