Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ scheduler = DDIMScheduler.from_pretrained(model, subfolder="scheduler")
|
|
10 |
pipe = DiffusionPipeline.from_pretrained(model, vae=vae, scheduler=scheduler, torch_dtype=torch.float16, use_safetensors=True, variant="fp16").to("cuda")
|
11 |
|
12 |
#pipe.enable_model_cpu_offload()
|
13 |
-
|
14 |
|
15 |
# Apply hidiffusion with a single line of code.
|
16 |
apply_hidiffusion(pipe)
|
|
|
10 |
pipe = DiffusionPipeline.from_pretrained(model, vae=vae, scheduler=scheduler, torch_dtype=torch.float16, use_safetensors=True, variant="fp16").to("cuda")
|
11 |
|
12 |
#pipe.enable_model_cpu_offload()
|
13 |
+
pipe.enable_vae_tiling()
|
14 |
|
15 |
# Apply hidiffusion with a single line of code.
|
16 |
apply_hidiffusion(pipe)
|