Spaces:

Menyu
/

NoobXL

Running on Zero

Menyu commited on Nov 7, 2024

Commit

dd03b5e

verified ·

1 Parent(s): 6cc69f4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,9 @@ 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)
     pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl",
         vae=vae,

 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("n0madic/MOHAWK_v20BakedVAE", torch_dtype=torch.float16)
     pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl",
         vae=vae,