Update app.py

app.py

@@ -13,8 +13,10 @@ from compel import Compel, ReturnedEmbeddingsType
 from safetensors.torch import load_file
 import spaces
 
-model_id = "aipicasso/emix-0-4-turbo"
+model_id = "aipicasso/emix-0-5"
 auth_token=os.environ["ACCESS_TOKEN"]
+repo = "ByteDance/SDXL-Lightning"
+ckpt = "sdxl_lightning_4step_lora.safetensors"
 
 scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler",token=auth_token)
 
@@ -24,6 +26,9 @@ pipe = AutoPipelineForText2Image.from_pretrained(
   scheduler=scheduler,
   token=auth_token)
 
+pipe.load_lora_weights(hf_hub_download(repo, ckpt))
+pipe.fuse_lora()
+
 pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.3, b2=1.4)
 
 state_dict = load_file("unaestheticXLv31.safetensors")
@@ -47,8 +52,8 @@ def inference(prompt, guidance, steps, seed=0, neg_prompt="", disable_auto_promp
 
   prompt,neg_prompt=auto_prompt_correction(prompt,neg_prompt,disable_auto_prompt_correction)
 
-  height=768
-  width=768
+  height=1024
+  width=1024
 
   print(prompt,neg_prompt)