Update requirements as per sd-scripts suggestion

bmaltais · Aug 27, 2024 · 0fd9c58 · 0fd9c58
1 parent de38d2a
commit 0fd9c58
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 11 deletions.
diff --git a/kohya_gui/flux_merge_lora_gui.py b/kohya_gui/flux_merge_lora_gui.py
@@ -330,7 +330,7 @@ def list_save_to(path):
                 )
                 save_precision = gr.Radio(
                     label="Save precision",
-                    choices=["float", "fp16", "bf16"],
+                    choices=["float", "fp16", "bf16", "fp8"],
                     value="fp16",
                     interactive=True,
                 )
@@ -420,11 +420,11 @@ def merge_flux_lora(
         lora_models = [model for model in models if model]
         ratios = [ratio for model, ratio in zip(models, [ratio_a, ratio_b, ratio_c, ratio_d]) if model]
 
-        if not verify_conditions(flux_model, lora_models):
-            log.info(
-                "Warning: Either provide at least one LoRA model along with the FLUX model or at least two LoRA models if no FLUX model is provided."
-            )
-            return
+        # if not verify_conditions(flux_model, lora_models):
+        #     log.info(
+        #         "Warning: Either provide at least one LoRA model along with the FLUX model or at least two LoRA models if no FLUX model is provided."
+        #     )
+        #     return
 
         for model in [flux_model] + lora_models:
             if not check_model(model):

diff --git a/requirements.txt b/requirements.txt
@@ -19,7 +19,7 @@ onnx==1.16.1
 prodigyopt==1.0
 protobuf==3.20.3
 open-clip-torch==2.20.0
-opencv-python==4.7.0.68
+opencv-python==4.7.0.72
 prodigyopt==1.0
 pytorch-lightning==1.9.0
 rich>=13.7.1

diff --git a/requirements_linux.txt b/requirements_linux.txt
@@ -1,4 +1,4 @@
-torch==2.1.2+cu118 torchvision==0.16.2+cu118 xformers==0.0.23.post1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+torch==2.4.0+cu118 torchvision==0.19.0+cu118 xformers==0.0.27.post2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
 bitsandbytes==0.43.3
 tensorboard==2.15.2 tensorflow==2.15.0.post1
 onnxruntime-gpu==1.17.1

diff --git a/requirements_pytorch_windows.txt b/requirements_pytorch_windows.txt
@@ -1,3 +1,3 @@
-torch==2.1.2+cu118 --index-url https://download.pytorch.org/whl/cu118
-torchvision==0.16.2+cu118 --index-url https://download.pytorch.org/whl/cu118
-xformers==0.0.23.post1+cu118 --index-url https://download.pytorch.org/whl/cu118
+torch==2.4.0+cu118 --index-url https://download.pytorch.org/whl/cu118
+torchvision==0.19.0+cu118 --index-url https://download.pytorch.org/whl/cu118
+xformers==0.0.27.post2+cu118 --index-url https://download.pytorch.org/whl/cu118