Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,12 +10,12 @@ from diffusers import KandinskyV22PriorPipeline, KandinskyV22Pipeline
|
|
10 |
|
11 |
accelerator = Accelerator(cpu=True)
|
12 |
|
13 |
-
generator = torch.Generator(device="cpu").manual_seed(
|
14 |
pope_prior = accelerator.prepare(KandinskyV22PriorPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-prior", torch_dtype=torch.float32))
|
15 |
-
pope_prior =
|
16 |
|
17 |
pope = accelerator.prepare(KandinskyV22Pipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float32))
|
18 |
-
pope =
|
19 |
|
20 |
def plex(img, cook, one, two, three):
|
21 |
goof = load_image(img).resize((512, 512))
|
|
|
10 |
|
11 |
accelerator = Accelerator(cpu=True)
|
12 |
|
13 |
+
generator = torch.Generator(device="cpu").manual_seed(4096)
|
14 |
pope_prior = accelerator.prepare(KandinskyV22PriorPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-prior", torch_dtype=torch.float32))
|
15 |
+
pope_prior = pope_prior.to("cpu")
|
16 |
|
17 |
pope = accelerator.prepare(KandinskyV22Pipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float32))
|
18 |
+
pope = pope.to("cpu")
|
19 |
|
20 |
def plex(img, cook, one, two, three):
|
21 |
goof = load_image(img).resize((512, 512))
|