Menyu commited on
Commit
af49643
·
verified ·
1 Parent(s): 60009a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -246,7 +246,7 @@ MAX_SEED = np.iinfo(np.int32).max
246
  MAX_IMAGE_SIZE = 2048
247
 
248
  if torch.cuda.is_available():
249
- vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
250
  token = os.environ.get("HF_TOKEN") # 从环境变量读取令牌
251
  model_path = hf_hub_download(
252
  repo_id="Menyu/MiaoMiao_Harem1.6", # 模型仓库名称(非完整URL)
@@ -255,7 +255,7 @@ if torch.cuda.is_available():
255
  )
256
  pipe = StableDiffusionXLPipeline.from_single_file(
257
  model_path,
258
- vae=vae,
259
  use_safetensors=True,
260
  torch_dtype=torch.float16,
261
  )
 
246
  MAX_IMAGE_SIZE = 2048
247
 
248
  if torch.cuda.is_available():
249
+ #vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
250
  token = os.environ.get("HF_TOKEN") # 从环境变量读取令牌
251
  model_path = hf_hub_download(
252
  repo_id="Menyu/MiaoMiao_Harem1.6", # 模型仓库名称(非完整URL)
 
255
  )
256
  pipe = StableDiffusionXLPipeline.from_single_file(
257
  model_path,
258
+ #vae=vae, 内置VAE
259
  use_safetensors=True,
260
  torch_dtype=torch.float16,
261
  )