Update package versions in cog.yaml for compatibility; comment out reference image optimization in predict.py; change deployment branch to 'dev' in deploy_public.yaml.

Author: Saquib764

.github/workflows/deploy_public.yaml

@@ -3,7 +3,7 @@ name: Push to TheFluxTrain public Replicate model
 on:
   push:
     branches:
-      - main
+      - dev
 
 jobs:
   push_to_replicate:

cog.yaml

@@ -17,18 +17,18 @@ build:
   # a list of packages in the format <package-name>==<version>
   python_packages:
     - diffusers==0.35.1
-    - transformers
+    - transformers==4.55.2
     - sentencepiece
     - protobuf
     - huggingface-hub==0.34.0
-    - peft
+    - peft==0.17.0
     - opencv-python
     - lightning
     - datasets
     - prodigyopt
     - wandb
     - Pillow==9.5.0
-    - accelerate==0.33.0
+    - accelerate==1.2.1
     - py_real_esrgan
     - rembg
     - onnxruntime

predict.py

@@ -136,8 +136,8 @@ def predict(
             print("has_reference: ", has_reference)
             print("reference_image: ", reference_image)
             print("delta: ", delta)
-            if has_reference:
-                optimised_reference, new_reference_delta = optimise_image_condition(reference_image, delta)
+            # if has_reference:
+            #     optimised_reference, new_reference_delta = optimise_image_condition(reference_image, delta)
             result_img = self.pipe(
                 prompt=prompt,
                 image=image,