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 |
|