pcuenq HF staff commited on
Commit
47b2c20
·
1 Parent(s): a1f905f

Update LoRA loading

Browse files
Files changed (1) hide show
  1. 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 = "lcm-sd/lcm-sd1.5-lora"
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