Spaces:

guangkaixu
/

GenPercept

Running on Zero

guangkaixu commited on Apr 7, 2024

Commit

eb1d34b

1 Parent(s): a123370

upload

Files changed (1) hide show

app.py CHANGED Viewed

@@ -266,7 +266,7 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    dtype = torch.float32
     vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae').to(dtype)
     unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_depth_v1").to(dtype)

     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    dtype = torch.float16
     vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae').to(dtype)
     unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_depth_v1").to(dtype)