Update app.py

app.py

@@ -38,7 +38,7 @@ taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).
 good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to(device)
 
 txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to(device)
-# txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
+txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
 
 # img2img model
 img2img_pipe = AutoPipelineForImage2Image.from_pretrained(base_model, vae=good_vae, transformer=txt2img_pipe.transformer, text_encoder=txt2img_pipe.text_encoder, tokenizer=txt2img_pipe.tokenizer, text_encoder_2=txt2img_pipe.text_encoder_2, tokenizer_2=txt2img_pipe.tokenizer_2, torch_dtype=dtype)
@@ -67,7 +67,7 @@ class calculateDuration:
         else:
             print(f"Elapsed time: {self.elapsed_time:.6f} seconds")
         
-@spaces.GPU(duration=120)
+
 def generate_image(orginal_image, prompt, adapter_names,  steps, seed, image_strength, cfg_scale, width, height, progress):
 
     
@@ -147,7 +147,8 @@ def upload_image_to_r2(image, account_id, access_key, secret_key, bucket_name):
 
 def generate_random_4_digit_string():
     return ''.join(random.choices(string.digits, k=4))
-    
+
+@spaces.GPU(duration=120)
 def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, steps, randomize_seed, seed, width, height, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
     print("run_lora", prompt, lora_strings_json, cfg_scale, steps, width, height)
     gr.Info("Starting process")
@@ -167,10 +168,6 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, s
     with calculateDuration("Unloading LoRA"):
         img2img_pipe.unload_lora_weights()
         txt2img_pipe.unload_lora_weights()
-        
-    print(txt2img_pipe.get_active_adapters())
-    list_adapters_component_wise = txt2img_pipe.get_list_adapters()
-    print(list_adapters_component_wise)
     
     lora_configs = None
     adapter_names = []