JoPmt commited on
Commit
1c8afdf
1 Parent(s): c85da47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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(43)
14
  pope_prior = accelerator.prepare(KandinskyV22PriorPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-prior", torch_dtype=torch.float32))
15
- pope_prior = accelerator.prepare(pope_prior.to("cpu"))
16
 
17
  pope = accelerator.prepare(KandinskyV22Pipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float32))
18
- pope = accelerator.prepare(pope.to("cpu"))
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))