Tanut commited on
Commit
b2d3d41
·
1 Parent(s): dcf7ed7
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -49,8 +49,8 @@ BASE_MODELS = {
49
  }
50
 
51
  # ControlNets
52
- # CN_QRMON = "monster-labs/control_v1p_sd15_qrcode_monster"
53
- CN_QRMON = "Nacholmo/controlnet-qr-pattern-v2"
54
  CN_BRIGHT = "latentcat/control_v1p_sd15_brightness"
55
 
56
  # ---------- helpers ----------
@@ -221,7 +221,7 @@ def get_qr_cn():
221
  global _CN_QR
222
  if _CN_QR is None:
223
  _CN_QR = ControlNetModel.from_pretrained(
224
- CN_QRMON, torch_dtype=DTYPE, use_safetensors=False, token=HF_TOKEN
225
  )
226
  return _CN_QR
227
 
 
49
  }
50
 
51
  # ControlNets
52
+ CN_QRMON = "monster-labs/control_v1p_sd15_qrcode_monster"
53
+ # CN_QRMON = "Nacholmo/controlnet-qr-pattern-v2"
54
  CN_BRIGHT = "latentcat/control_v1p_sd15_brightness"
55
 
56
  # ---------- helpers ----------
 
221
  global _CN_QR
222
  if _CN_QR is None:
223
  _CN_QR = ControlNetModel.from_pretrained(
224
+ CN_QRMON, torch_dtype=DTYPE, use_safetensors=True, token=HF_TOKEN
225
  )
226
  return _CN_QR
227