Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ IMAGE_SIZE = 1024
|
|
37 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
38 |
base_model = "black-forest-labs/FLUX.1-dev"
|
39 |
|
40 |
-
controlnet_model = '
|
41 |
controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
|
42 |
|
43 |
|
@@ -207,7 +207,7 @@ def load_loras(lora_strings_json:str):
|
|
207 |
if lora_configs:
|
208 |
|
209 |
with calculateDuration("Loading LoRA weights"):
|
210 |
-
active_adapters = pipe.get_active_adapters()
|
211 |
adapter_names = []
|
212 |
adapter_weights = []
|
213 |
for lora_info in lora_configs:
|
@@ -220,9 +220,9 @@ def load_loras(lora_strings_json:str):
|
|
220 |
adapter_names.append(adapter_name)
|
221 |
adapter_weights.append(adapter_weight)
|
222 |
|
223 |
-
if adapter_name in active_adapters:
|
224 |
-
|
225 |
-
|
226 |
|
227 |
try:
|
228 |
pipe.load_lora_weights(lora_repo, weight_name=weights, adapter_name=adapter_name)
|
@@ -231,8 +231,8 @@ def load_loras(lora_strings_json:str):
|
|
231 |
continue
|
232 |
|
233 |
# set lora weights
|
234 |
-
if len(adapter_names) > 0:
|
235 |
-
|
236 |
|
237 |
|
238 |
def generate_control_image(image, mask, width, height):
|
|
|
37 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
38 |
base_model = "black-forest-labs/FLUX.1-dev"
|
39 |
|
40 |
+
controlnet_model = 'black-forest-labs/FLUX.1-Canny-dev'
|
41 |
controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
|
42 |
|
43 |
|
|
|
207 |
if lora_configs:
|
208 |
|
209 |
with calculateDuration("Loading LoRA weights"):
|
210 |
+
#active_adapters = pipe.get_active_adapters()
|
211 |
adapter_names = []
|
212 |
adapter_weights = []
|
213 |
for lora_info in lora_configs:
|
|
|
220 |
adapter_names.append(adapter_name)
|
221 |
adapter_weights.append(adapter_weight)
|
222 |
|
223 |
+
#if adapter_name in active_adapters:
|
224 |
+
# print(f"Adapter '{adapter_name}' is already loaded, skipping.")
|
225 |
+
# continue
|
226 |
|
227 |
try:
|
228 |
pipe.load_lora_weights(lora_repo, weight_name=weights, adapter_name=adapter_name)
|
|
|
231 |
continue
|
232 |
|
233 |
# set lora weights
|
234 |
+
#if len(adapter_names) > 0:
|
235 |
+
# pipe.set_adapters(adapter_names, adapter_weights=adapter_weights)
|
236 |
|
237 |
|
238 |
def generate_control_image(image, mask, width, height):
|