Update LoRA loading
Browse files- app-controlnetlora.py +1 -3
app-controlnetlora.py
CHANGED
@@ -77,7 +77,7 @@ controlnet_canny = ControlNetModel.from_pretrained(
|
|
77 |
canny_torch = SobelOperator(device=device)
|
78 |
|
79 |
model_id = "nitrosocke/mo-di-diffusion"
|
80 |
-
lcm_lora_id = "
|
81 |
|
82 |
if SAFETY_CHECKER == "True":
|
83 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
@@ -103,8 +103,6 @@ if psutil.virtual_memory().total < 64 * 1024**3:
|
|
103 |
# Load LCM LoRA
|
104 |
pipe.load_lora_weights(
|
105 |
lcm_lora_id,
|
106 |
-
weight_name="lcm_sd_lora.safetensors",
|
107 |
-
adapter_name="lcm",
|
108 |
use_auth_token=HF_TOKEN,
|
109 |
)
|
110 |
|
|
|
77 |
canny_torch = SobelOperator(device=device)
|
78 |
|
79 |
model_id = "nitrosocke/mo-di-diffusion"
|
80 |
+
lcm_lora_id = "latent-consistency/lcm-lora-sdv1-5"
|
81 |
|
82 |
if SAFETY_CHECKER == "True":
|
83 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
|
|
103 |
# Load LCM LoRA
|
104 |
pipe.load_lora_weights(
|
105 |
lcm_lora_id,
|
|
|
|
|
106 |
use_auth_token=HF_TOKEN,
|
107 |
)
|
108 |
|