Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
from diffusers import UniDiffuserPipeline
|
4 |
from diffusers.utils import load_image
|
5 |
from accelerate import Accelerator
|
6 |
-
|
7 |
|
8 |
pipe = accelerator.prepare(UniDiffuserPipeline.from_pretrained("thu-ml/unidiffuser-v1", torch_dtype=torch.bfloat16))
|
9 |
pipe = accelerator.prepare(pipe.to("cpu"))
|
|
|
3 |
from diffusers import UniDiffuserPipeline
|
4 |
from diffusers.utils import load_image
|
5 |
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"))
|