Menyu commited on
Commit
068b169
·
verified ·
1 Parent(s): 18882fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -246,7 +246,6 @@ 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/Pixel", # 模型仓库名称(非完整URL)
@@ -255,7 +254,6 @@ 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
  token = os.environ.get("HF_TOKEN") # 从环境变量读取令牌
250
  model_path = hf_hub_download(
251
  repo_id="Menyu/Pixel", # 模型仓库名称(非完整URL)
 
254
  )
255
  pipe = StableDiffusionXLPipeline.from_single_file(
256
  model_path,
 
257
  use_safetensors=True,
258
  torch_dtype=torch.float16,
259
  )