Update app.py

app.py

@@ -118,7 +118,7 @@ pipe = StableDiffusionXLInstantIDImg2ImgPipeline.from_pretrained("rubbrband/albe
                                                                  torch_dtype=torch.float16)
 
 compel = Compel(tokenizer=[pipe.tokenizer, pipe.tokenizer_2] , text_encoder=[pipe.text_encoder, pipe.text_encoder_2], returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED, requires_pooled=[False, True])
-
+compel.to(device)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
 pipe.load_ip_adapter_instantid(face_adapter)
 pipe.set_ip_adapter_scale(0.8)
@@ -205,7 +205,6 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
                 )
                 del weights_sd
                 del lora_model
-                gc.collect()
 @spaces.GPU
 def generate_image(prompt, negative, face_emb, face_image, image_strength, images, guidance_scale, face_strength, depth_control_scale):
     conditioning, pooled = compel(prompt)
@@ -307,7 +306,6 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
 
     image = generate_image(prompt, negative, face_emb, face_image, image_strength, images, guidance_scale, face_strength, depth_control_scale)
     last_lora = repo_name
-    gc.collect()
     return image, gr.update(visible=True)
 
 def shuffle_gallery(sdxl_loras):