Anonymous commited on
Commit
1b39655
·
1 Parent(s): b39a2ed

load in advance

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from free_lunch_utils import register_free_upblock2d, register_free_crossattn_up
9
  from pipeline_freescale import StableDiffusionXLPipeline
10
  from pipeline_freescale_turbo import StableDiffusionXLPipeline_Turbo
11
 
12
- dtype = torch.bfloat16
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  model_ckpt = "stabilityai/stable-diffusion-xl-base-1.0"
15
  model_ckpt_turbo = "stabilityai/sdxl-turbo"
 
9
  from pipeline_freescale import StableDiffusionXLPipeline
10
  from pipeline_freescale_turbo import StableDiffusionXLPipeline_Turbo
11
 
12
+ dtype = torch.float16
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  model_ckpt = "stabilityai/stable-diffusion-xl-base-1.0"
15
  model_ckpt_turbo = "stabilityai/sdxl-turbo"