Changed transformers version.

requirements.txt

@@ -3,7 +3,7 @@ numpy==1.26.4
 pandas==2.2.2
 phonemizer==3.3.0
 encodec==0.1.1
-transformers
+transformers==4.45.2
 torch==2.1.0
 torchaudio==2.1.0
 torchvision==0.16.0

src/model/modules/imagecraft.py

@@ -90,7 +90,6 @@ class ImageCraft(nn.Module):
 
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
 
-    # Copied from transformers.models.llava.modeling_llava.LlavaForConditionalGeneration.tie_weights with Llava->PaliGemma
     def tie_weights(self):
         return self.language_model.tie_weights()