Update evo_ukiyoe_v1.py

evo_ukiyoe_v1.py

@@ -18,8 +18,8 @@ DPO_REPO = "mhdang/dpo-sdxl-text2image-v1"
 JN_REPO = "RunDiffusion/Juggernaut-XL-v9"
 JSDXL_REPO = "stabilityai/japanese-stable-diffusion-xl"
 
-# LoRA weights
-LORA_REPO = "SakanaAI/Evo-Ukiyoe-v1"
+# Evo-Ukiyoe
+UKIYOE_REPO = "SakanaAI/Evo-Ukiyoe-v1"
 
 
 def load_state_dict(checkpoint_file: Union[str, os.PathLike], device: str = "cpu"):
@@ -164,8 +164,8 @@ def load_evo_ukiyoe(device="cuda") -> StableDiffusionXLPipeline:
     )
     pipe = pipe.to(device, dtype=torch.float16)
 
-    # Load LoRA module
-    pipe.load_lora_weights(LORA_REPO)
+    # Load Evo-Ukiyoe weights
+    pipe.load_lora_weights(UKIYOE_REPO)
     pipe.fuse_lora(lora_scale=1.0)
     return pipe