Keltezaa commited on
Commit
251bee7
·
verified ·
1 Parent(s): 098f460

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -469,10 +469,13 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
469
  appends.append(trigger_word)
470
  prompt_mash = " ".join(prepends + [prompt] + appends)
471
  print("Prompt Mash: ", prompt_mash)
 
472
 
473
  # Unload previous LoRA weights
474
  with calculateDuration("Unloading LoRA"):
475
  pipe.unload_lora_weights()
 
 
476
 
477
  # Load LoRA weights
478
  lora_names = []
@@ -481,6 +484,7 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
481
  for idx, lora in enumerate(selected_loras):
482
  lora_name = f"lora_{idx}"
483
  lora_names.append(lora_name)
 
484
  lora_weights.append(lora_scale_1 if idx == 0 else lora_scale_2)
485
  pipe.load_lora_weights(
486
  lora['repo'],
@@ -488,6 +492,10 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
488
  low_cpu_mem_usage=True,
489
  adapter_name=lora_name,
490
  )
 
 
 
 
491
  pipe.set_adapters(lora_names, adapter_weights=lora_weights)
492
 
493
  # Set random seed if required
@@ -584,7 +592,7 @@ css = '''
584
  #component-11{align-self: stretch;}
585
  '''
586
  font=[gr.themes.GoogleFont("Source Sans Pro"), "Arial", "sans-serif"]
587
- with gr.Blocks(theme=gr.themes.Soft(font=font), css=css, delete_cache=(256, 256)) as app:
588
  title = gr.HTML(
589
  """<h1><img src="https://huggingface.co/spaces/keltezaa/Celebrity_pony_NSFW_Mix/resolve/main/solo-traveller_16875043.png" alt="LoRA"> Celebrity Pony NSFW Mix</h1>""",
590
  elem_id="title",
 
469
  appends.append(trigger_word)
470
  prompt_mash = " ".join(prepends + [prompt] + appends)
471
  print("Prompt Mash: ", prompt_mash)
472
+ print("--Seed--:", seed)
473
 
474
  # Unload previous LoRA weights
475
  with calculateDuration("Unloading LoRA"):
476
  pipe.unload_lora_weights()
477
+
478
+ print(pipe.get_active_adapters())
479
 
480
  # Load LoRA weights
481
  lora_names = []
 
484
  for idx, lora in enumerate(selected_loras):
485
  lora_name = f"lora_{idx}"
486
  lora_names.append(lora_name)
487
+ print(f"Lora Name: {lora_name}")
488
  lora_weights.append(lora_scale_1 if idx == 0 else lora_scale_2)
489
  pipe.load_lora_weights(
490
  lora['repo'],
 
492
  low_cpu_mem_usage=True,
493
  adapter_name=lora_name,
494
  )
495
+
496
+ print("Loaded LoRAs:", lora_names)
497
+ print("Adapter weights:", lora_weights)
498
+
499
  pipe.set_adapters(lora_names, adapter_weights=lora_weights)
500
 
501
  # Set random seed if required
 
592
  #component-11{align-self: stretch;}
593
  '''
594
  font=[gr.themes.GoogleFont("Source Sans Pro"), "Arial", "sans-serif"]
595
+ with gr.Blocks(theme=gr.themes.Soft(font=font), css=css, delete_cache=(128, 256)) as app:
596
  title = gr.HTML(
597
  """<h1><img src="https://huggingface.co/spaces/keltezaa/Celebrity_pony_NSFW_Mix/resolve/main/solo-traveller_16875043.png" alt="LoRA"> Celebrity Pony NSFW Mix</h1>""",
598
  elem_id="title",