ford442 commited on
Commit
6731d9e
·
verified ·
1 Parent(s): c819219

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained("ford442/stable-diffusion-3.5-me
76
  #pipe = torch.compile(pipe)
77
  # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
78
 
79
- refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("ford442/stable-diffusion-xl-refiner-1.0-bf16", vaeXL, requires_aesthetics_score=True) #.to(torch.bfloat16)
80
  #refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float32, requires_aesthetics_score=True, device_map='balanced')
81
  refiner.scheduler=EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config)
82
  #refiner.enable_model_cpu_offload()
 
76
  #pipe = torch.compile(pipe)
77
  # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
78
 
79
+ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("ford442/stable-diffusion-xl-refiner-1.0-bf16",vae = vaeXL, requires_aesthetics_score=True) #.to(torch.bfloat16)
80
  #refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float32, requires_aesthetics_score=True, device_map='balanced')
81
  refiner.scheduler=EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config)
82
  #refiner.enable_model_cpu_offload()