JoPmt commited on
Commit
4a0a5e6
1 Parent(s): 544fc1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from accelerate import Accelerator
6
  accelerator = Accelerator(cpu=True)
7
 
8
  pipe = accelerator.prepare(UniDiffuserPipeline.from_pretrained("thu-ml/unidiffuser-v1", torch_dtype=torch.bfloat16))
9
- pipe = accelerator.prepare(pipe.to("cpu"))
10
  apol=[]
11
  def plex(image_url,stips):
12
  init_image = load_image(image_url).resize((512, 512))
 
6
  accelerator = Accelerator(cpu=True)
7
 
8
  pipe = accelerator.prepare(UniDiffuserPipeline.from_pretrained("thu-ml/unidiffuser-v1", torch_dtype=torch.bfloat16))
9
+ pipe = pipe.to("cpu")
10
  apol=[]
11
  def plex(image_url,stips):
12
  init_image = load_image(image_url).resize((512, 512))