Spaces:

briaai
/

BRIA-2.2-ControlNet-Depth

Runtime error

dvir-bria commited on Jun 6

Commit

1cb6298

•

1 Parent(s): e1576c8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     device_map='auto',
     low_cpu_mem_usage=True,
     offload_state_dict=True,
-).to('cuda')
 pipe.scheduler = EulerAncestralDiscreteScheduler(
     beta_start=0.00085,
     beta_end=0.012,

     device_map='auto',
     low_cpu_mem_usage=True,
     offload_state_dict=True,
+).to('cuda').to(torch.float16)
 pipe.scheduler = EulerAncestralDiscreteScheduler(
     beta_start=0.00085,
     beta_end=0.012,