ford442 commited on
Commit
5330b01
·
verified ·
1 Parent(s): bad1ff2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,11 +72,11 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("ford442/stable-diffu
72
  #refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float32, requires_aesthetics_score=True, device_map='balanced')
73
 
74
  #refiner.enable_model_cpu_offload()
75
- refiner.scheduler = EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config, beta_schedule="scaled_linear")
76
 
77
  #refiner.scheduler.config.requires_aesthetics_score=False
78
  refiner.to(device)
79
  #refiner = torch.compile(refiner)
 
80
 
81
  tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False, device_map='balanced')
82
  tokenizer.tokenizer_legacy=False
 
72
  #refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float32, requires_aesthetics_score=True, device_map='balanced')
73
 
74
  #refiner.enable_model_cpu_offload()
 
75
 
76
  #refiner.scheduler.config.requires_aesthetics_score=False
77
  refiner.to(device)
78
  #refiner = torch.compile(refiner)
79
+ refiner.scheduler = EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config, beta_schedule="scaled_linear")
80
 
81
  tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False, device_map='balanced')
82
  tokenizer.tokenizer_legacy=False