thewhole commited on
Commit
c5152a3
1 Parent(s): 981e791

Update shape.py

Browse files
Files changed (1) hide show
  1. shape.py +1 -1
shape.py CHANGED
@@ -23,7 +23,7 @@ pipe_kwargs = {
23
  "requires_safety_checker": False,
24
  "torch_dtype": torch.float16 ,
25
  }
26
- self.pipe = StableDiffusionPipeline.from_pretrained(
27
  "stabilityai/stable-diffusion-2-1-base",
28
  **pipe_kwargs,
29
  )
 
23
  "requires_safety_checker": False,
24
  "torch_dtype": torch.float16 ,
25
  }
26
+ pipe = StableDiffusionPipeline.from_pretrained(
27
  "stabilityai/stable-diffusion-2-1-base",
28
  **pipe_kwargs,
29
  )