Update modeling_pixel.py

modeling_pixel.py

@@ -19,18 +19,20 @@ class ResidualBlock(nn.Module):
 
 class TopAIImageGenerator(PreTrainedModel):
     config_class = TopAIImageConfig
-    # שימוש ברשימה ריקה כסטנדרט של Transformers
-    all_tied_weights_keys = [] 
+    
+    # תיקון ה-AttributeError: חייב להיות מילון (dict) כדי שתהיה לו מתודה .keys()
+    all_tied_weights_keys = {} 
 
     def __init__(self, config):
         super().__init__(config)
-        h = config.hidden_dim # מקבל 512 מה-config.json
+        # שימוש ב-hidden_dim מהקונפיג (512)
+        h = config.hidden_dim 
         
         self.text_projection = nn.Linear(config.input_dim, 4 * 4 * h)
         
-        # בנייה דינמית שמתאימה למשקולות שאומנו ב-512
+        # בנייה דינמית שמתאימה בדיוק למשקולות ב-Safetensors
         self.decoder = nn.Sequential(
-            # שכבה 0: מ-512 ל-512 (כאן הייתה השגיאה!)
+            # שכבה 0: מ-512 ל-512 (כאן היה ה-Mismatch)
             self._upsample(h, h), 
             # שכבה 1
             ResidualBlock(h),
@@ -59,7 +61,7 @@ class TopAIImageGenerator(PreTrainedModel):
         )
 
     def forward(self, text_embeddings):
-        # המרת ה-Embedding למפה של 4x4
+        # שינוי צורה למפת מאפיינים ראשונית
         x = self.text_projection(text_embeddings)
         x = x.view(-1, self.config.hidden_dim, 4, 4)
         return self.decoder(x)