Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import gc
|
|
| 10 |
# โโโ Config โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
|
| 11 |
MODEL_ID = "Laxhar/noobai-XL-1.1" # HF repo (public, diffusers-compatible)
|
| 12 |
DEVICE = "cpu"
|
| 13 |
-
DTYPE = torch.
|
| 14 |
DEFAULT_W, DEFAULT_H = 640, 960
|
| 15 |
DEFAULT_STEPS = 20
|
| 16 |
DEFAULT_GUIDANCE = 5.0
|
|
@@ -93,7 +93,7 @@ def get_pipeline():
|
|
| 93 |
if _pipe is not None:
|
| 94 |
return _pipe
|
| 95 |
|
| 96 |
-
print("Loading NoobAI-XL SDNQ pipeline on CPU (
|
| 97 |
|
| 98 |
# 1) VAE (ๅ็ฌๅ ่ฝฝ๏ผๅฏ็จ slicing)
|
| 99 |
vae = AutoencoderKL.from_pretrained(
|
|
|
|
| 10 |
# โโโ Config โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
|
| 11 |
MODEL_ID = "Laxhar/noobai-XL-1.1" # HF repo (public, diffusers-compatible)
|
| 12 |
DEVICE = "cpu"
|
| 13 |
+
DTYPE = torch.bfloat16 # CPU ไธๆฏๆ fp16 ่ฟ็ฎ๏ผๆน็จ bfloat16๏ผๅๆ ท 2 ๅญ่/ๅๆฐ๏ผ็ๅ
ๅญไธ CPU ๅ็ๆฏๆ๏ผ
|
| 14 |
DEFAULT_W, DEFAULT_H = 640, 960
|
| 15 |
DEFAULT_STEPS = 20
|
| 16 |
DEFAULT_GUIDANCE = 5.0
|
|
|
|
| 93 |
if _pipe is not None:
|
| 94 |
return _pipe
|
| 95 |
|
| 96 |
+
print("Loading NoobAI-XL SDNQ pipeline on CPU (bf16)...")
|
| 97 |
|
| 98 |
# 1) VAE (ๅ็ฌๅ ่ฝฝ๏ผๅฏ็จ slicing)
|
| 99 |
vae = AutoencoderKL.from_pretrained(
|