Update configuration_ostlm.py

configuration_ostlm.py

@@ -14,10 +14,17 @@ class OSTLMConfig(PretrainedConfig):
         dropout=0.1,
         pad_token_id=0,
         decoder_start_token_id=101,
-        is_encoder_decoder=True, # חשוב מאוד! אומר לספרייה שזה מודל תרגום
+        eos_token_id=102,
         **kwargs
     ):
-        super().__init__(pad_token_id=pad_token_id, decoder_start_token_id=decoder_start_token_id, **kwargs)
+        # הגדרה כמודל Encoder-Decoder כדי להפעיל את לוגיקת התרגום
+        super().__init__(
+            pad_token_id=pad_token_id, 
+            decoder_start_token_id=decoder_start_token_id, 
+            eos_token_id=eos_token_id,
+            is_encoder_decoder=True, 
+            **kwargs
+        )
         self.vocab_size = vocab_size
         self.d_model = d_model
         self.nhead = nhead
@@ -26,6 +33,5 @@ class OSTLMConfig(PretrainedConfig):
         self.dim_feedforward = dim_feedforward
         self.dropout = dropout
         
-        # השורות שפותרות את ה-AttributeError:
-        self.num_hidden_layers = num_decoder_layers 
-        self.is_encoder_decoder = True
+        # שדה חובה ש-Hugging Face מחפש ב-generate
+        self.num_hidden_layers = num_decoder_layers