Hugging Face's logo

Spaces:
Neurify
/
IllusionDiffusion
Runtime error

App Files Community
1
Neurify commited on
Commit
70cebc6
·
verified ·
1 Parent(s): feb58f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ from transformers import CLIPImageProcessor
26
  from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
27
 
28
  BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
29
- BASE_MODEL = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
30
 
31
  # Initialize both pipelines
32
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
 
26
  from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
27
 
28
  BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
29
+ # BASE_MODEL = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
30
 
31
  # Initialize both pipelines
32
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)