update

app.py

@@ -12,8 +12,11 @@ import gradio as gr
 import numpy as np
 
 device = "cuda"
-ckpt_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors")
-ckpt_IPA_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors-IP-Adapter-Plus")
+device = "cuda"
+ckpt_dir = '/home/lixiang46/Kolors/weights/Kolors'
+ckpt_IPA_dir = '/home/lixiang46/Kolors/weights/Kolors-IP-Adapter-Plus'
+# ckpt_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors")
+# ckpt_IPA_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors-IP-Adapter-Plus")
 
 # Load models
 text_encoder = ChatGLMModel.from_pretrained(f'{ckpt_dir}/text_encoder', torch_dtype=torch.float16).half().to(device)
@@ -60,8 +63,7 @@ def infer(prompt, ip_adapter_image, ip_adapter_scale, negative_prompt, seed, ran
             num_inference_steps=num_inference_steps, 
             guidance_scale=guidance_scale,
             num_images_per_prompt=1,
-            generator=generator,
-            image_encoder=image_encoder,
+            generator=generator
         ).images[0]
     
     return image
@@ -173,4 +175,4 @@ with gr.Blocks(css=css) as demo:
         outputs = [result]
     )
 
-demo.queue().launch()
+demo.queue().launch(share=True)