jiuface commited on
Commit
ba482d1
·
verified ·
1 Parent(s): 9a9c57b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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 = 'InstantX/FLUX.1-dev-Controlnet-Canny'
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
- 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,8 +231,8 @@ def load_loras(lora_strings_json:str):
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):
 
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):