juxuan27 commited on
Commit
2ba0d25
1 Parent(s): 85eb89a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,9 +49,9 @@ caption="A cake on the table."
49
  # conditioning scale
50
  paintingnet_conditioning_scale=1.0
51
 
52
- brushnet = BrushNetModel.from_pretrained(brushnet_path, torch_dtype=torch.float16)
53
  pipe = StableDiffusionBrushNetPipeline.from_pretrained(
54
- base_model_path, brushnet=brushnet, torch_dtype=torch.float16, low_cpu_mem_usage=False
55
  )
56
 
57
  # speed up diffusion process with faster scheduler and memory optimization
 
49
  # conditioning scale
50
  paintingnet_conditioning_scale=1.0
51
 
52
+ brushnet = BrushNetModel.from_pretrained(brushnet_path, torch_dtype=torch.float32)
53
  pipe = StableDiffusionBrushNetPipeline.from_pretrained(
54
+ base_model_path, brushnet=brushnet, torch_dtype=torch.float32, low_cpu_mem_usage=False
55
  )
56
 
57
  # speed up diffusion process with faster scheduler and memory optimization