Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
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))
|