Spaces:

K00B404
/

ipadapter_testing

Running

K00B404 commited on Apr 22

Commit

29d319f

verified ·

1 Parent(s): 7b9c473

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ noise_scheduler = DDIMScheduler(
     set_alpha_to_one=False,
     steps_offset=1,
 )
-vae = AutoencoderKL.from_pretrained(vae_model_path).to(dtype=torch.float16)
 def image_grid(imgs, rows, cols):
     assert len(imgs) == rows * cols

     set_alpha_to_one=False,
     steps_offset=1,
 )
+vae = AutoencoderKL.from_pretrained(vae_model_path)#.to(dtype=torch.float16)
 def image_grid(imgs, rows, cols):
     assert len(imgs) == rows * cols