Update modeling_pixel.py

modeling_pixel.py

@@ -18,21 +18,21 @@ class ResidualBlock(nn.Module):
 
 class TopAIImageGenerator(PreTrainedModel):
     config_class = TopAIImageConfig
-    all_tied_weights_keys = [] # מונע את ה-AttributeError ב-Transformers
+    # תיקון השגיאה: מגדירים כרשימה ריקה, Transformers כבר יסתדר עם זה
+    all_tied_weights_keys = [] 
 
     def __init__(self, config):
         super().__init__(config)
-        # תיקון המימד ל-4*4*512 בהתאם ל-Checkpoint שלכם
+        # סינכרון המימדים לפי ה-Checkpoint שלכם
         self.text_projection = nn.Linear(config.input_dim, 4 * 4 * config.hidden_dim)
         
         self.decoder = nn.Sequential(
-            self._upsample(config.hidden_dim, 256), # 4 -> 8
+            self._upsample(config.hidden_dim, 256), # 4x4 -> 8x8
             ResidualBlock(256),
-            self._upsample(256, 128),              # 8 -> 16
+            self._upsample(256, 128),              # 8x8 -> 16x16
             ResidualBlock(128),
-            self._upsample(128, 64),               # 16 -> 32
-            self._upsample(64, 32),                # 32 -> 64
-            self._upsample(32, 32),                # 64 -> 128 (תיקון למימד עקבי)
+            self._upsample(128, 64),               # 16x16 -> 32x32
+            self._upsample(64, 32),                # 32x32 -> 64x64
             nn.Conv2d(32, config.image_channels, kernel_size=3, padding=1),
             nn.Tanh()
         )