liyy201912 commited on
Commit
021be0e
Β·
1 Parent(s): 55d0d55

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. aagenerator.py +1 -1
aagenerator.py CHANGED
@@ -597,7 +597,7 @@ class Generator:
597
  # args.pretrained_model_name_or_path, subfolder="text_encoder", revision=args.revision
598
  # )
599
  self.vae = AutoencoderKL.from_pretrained(
600
- "/fsx_laion/alvin/pretrain/sd-vae-ft-mse"
601
  )
602
 
603
  vae_path = (
 
597
  # args.pretrained_model_name_or_path, subfolder="text_encoder", revision=args.revision
598
  # )
599
  self.vae = AutoencoderKL.from_pretrained(
600
+ "stabilityai/sd-vae-ft-mse"
601
  )
602
 
603
  vae_path = (