Tanut
commited on
Commit
·
b2d3d41
1
Parent(s):
dcf7ed7
Rollback
Browse files
app.py
CHANGED
|
@@ -49,8 +49,8 @@ BASE_MODELS = {
|
|
| 49 |
}
|
| 50 |
|
| 51 |
# ControlNets
|
| 52 |
-
|
| 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=
|
| 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 |
|