Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|