Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=t
|
|
42 |
|
43 |
|
44 |
pipe = FluxControlNetInpaintPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16).to(device)
|
45 |
-
pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
|
46 |
|
47 |
# pipe.enable_model_cpu_offload() # for saving memory
|
48 |
|
|
|
42 |
|
43 |
|
44 |
pipe = FluxControlNetInpaintPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16).to(device)
|
45 |
+
# pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
|
46 |
|
47 |
# pipe.enable_model_cpu_offload() # for saving memory
|
48 |
|