jacky3102 commited on
Commit
1084b8c
ยท
verified ยท
1 Parent(s): 00a8485

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.float16 # fp16 ๆ›ด็œๅ†…ๅญ˜๏ผŒcpu-basic (16GB) ๅฟ…้กป็”จ fp16
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 (fp16)...")
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(