dvir-bria commited on
Commit
1cb6298
1 Parent(s): e1576c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
20
  device_map='auto',
21
  low_cpu_mem_usage=True,
22
  offload_state_dict=True,
23
- ).to('cuda')
24
  pipe.scheduler = EulerAncestralDiscreteScheduler(
25
  beta_start=0.00085,
26
  beta_end=0.012,
 
20
  device_map='auto',
21
  low_cpu_mem_usage=True,
22
  offload_state_dict=True,
23
+ ).to('cuda').to(torch.float16)
24
  pipe.scheduler = EulerAncestralDiscreteScheduler(
25
  beta_start=0.00085,
26
  beta_end=0.012,