scales-optional-loras

This commit is contained in:
gokayfem
2024-10-18 02:18:20 +03:00
parent b92957021f
commit 331dcfe37f
+1 -1
View File
@@ -504,7 +504,7 @@ def create_blank_image(self):
return (img_tensor,)
# Add common methods to all classes
for cls in [FluxPro, FluxDev, FluxSchnell, FluxPro11, FluxGeneral]:
for cls in [FluxPro, FluxDev, FluxSchnell, FluxPro11, FluxGeneral, FluxLora]:
cls.process_result = process_result
cls.create_blank_image = create_blank_image