NoobXL11

Running on Zero

Menyu commited on Nov 7

Commit

e8658d9

•

1 Parent(s): 0bcd366

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,8 +13,8 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096
 if torch.cuda.is_available():
-    # vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-    vae = AutoencoderKL.from_pretrained("https://huggingface.co/scrapware/personal-backup/resolve/main/bakedvae/anyloraCheckpoint_bakedvaeTwinkleFp16.safetensors", torch_dtype=torch.float16)
     pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl",

 MAX_IMAGE_SIZE = 4096
 if torch.cuda.is_available():
+    vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+    #vae = AutoencoderKL.from_pretrained("https://huggingface.co/scrapware/personal-backup/resolve/main/bakedvae/anyloraCheckpoint_bakedvaeTwinkleFp16.safetensors", torch_dtype=torch.float16)
     pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl",