multimodalart HF staff commited on
Commit
577e167
·
verified ·
1 Parent(s): 34cb1b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -6,12 +6,10 @@ from safetensors.torch import load_file
6
  import spaces
7
 
8
  ### SDXL Turbo ####
9
-
10
  pipe_turbo = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
11
  pipe_turbo.to("cuda")
12
 
13
  ### SDXL Lightning ###
14
-
15
  base = "stabilityai/stable-diffusion-xl-base-1.0"
16
  repo = "ByteDance/SDXL-Lightning"
17
  ckpt = "sdxl_lightning_1step_unet_x0.safetensors"
 
6
  import spaces
7
 
8
  ### SDXL Turbo ####
 
9
  pipe_turbo = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
10
  pipe_turbo.to("cuda")
11
 
12
  ### SDXL Lightning ###
 
13
  base = "stabilityai/stable-diffusion-xl-base-1.0"
14
  repo = "ByteDance/SDXL-Lightning"
15
  ckpt = "sdxl_lightning_1step_unet_x0.safetensors"