Fabrice-TIERCELIN commited on
Commit
7a3f32b
1 Parent(s): 9b938b8

Fix warning

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import modin.pandas as pd
13
  max_64_bit_int = 2**63 - 1
14
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
- pipe = StableDiffusionXLInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", safety_checker = None)
17
  pipe = pipe.to(device)
18
 
19
  def noise_color(color, noise):
 
13
  max_64_bit_int = 2**63 - 1
14
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
+ pipe = StableDiffusionXLInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0")
17
  pipe = pipe.to(device)
18
 
19
  def noise_color(color, noise):