WillHeld
/

DiVA-llama-3-v0-8b

Feature Extraction

Model card Files Files and versions Community

Helw150 commited on 26 days ago

Commit

79fec82

•

1 Parent(s): 26bf120

Fix

Files changed (1) hide show

modeling_diva.py +1 -1

modeling_diva.py CHANGED Viewed

@@ -313,7 +313,7 @@ class DiVAModel(PreTrainedModel):
             )
             prefix = torch.cat(
                 [self.pre_system, user_prompt_text, self.post_system],
-                axis=1,
             )
         else:
             prefix = self.prefix

             )
             prefix = torch.cat(
                 [self.pre_system, user_prompt_text, self.post_system],
+                axis=0,
             )
         else:
             prefix = self.prefix