Upload modeling_mic21.py with huggingface_hub

modeling_mic21.py

@@ -47,7 +47,7 @@ class MIC21SummarizerModel(PreTrainedModel):
         #self.out_device = config.out_device
 
         #self.projection_layer = torch.nn.Linear(49, self.components["llm"].config.hidden_size, dtype=torch.float, device=f"cuda:{self.in_device}")
-        self.projection_layer = torch.nn.Linear(49, self.components["llm"].config.hidden_size, dtype=torch.float)
+        self.projection_layer = torch.nn.Linear(49, 2048, dtype=torch.float)
         
         #self.projection_norm = torch.nn.LayerNorm(49, eps=1e-5, bias=True, device=f"cuda:{self.in_device}")
         self.projection_norm = torch.nn.LayerNorm(49, eps=1e-5, bias=True)