Spaces:

briaai
/

BRIA-2.2-ControlNet-Recoloring

Running on Zero

dvir-bria commited on Jun 6

Commit

044f8b8

•

1 Parent(s): 20f5a2b

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,