Spaces:

alexeyGod
/

sdxl_turbo2

Running

alexeyGod commited on 24 days ago

Commit

90a23cb

verified ·

1 Parent(s): 06d24b9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
 pipe.vae = AutoencoderTiny.from_pretrained(
     "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
 pipe = pipe.to(device)
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
     prompt,

 pipe.vae = AutoencoderTiny.from_pretrained(
     "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
 pipe = pipe.to(device)
+pipe.enable_vae_tiling()
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
     prompt,