Spaces:

Chaerin5
/

FoundHand

Running on Zero

Chaerin5 commited on Dec 28, 2024

Commit

6c08e51

1 Parent(s): 10cf900

enable zerogpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -219,7 +219,7 @@ if NEW_MODEL:
     vae_state_dict = torch.load(vae_path,map_location='cpu')['state_dict']
     autoencoder = vqvae.create_model(3, 3, opts.latent_dim).eval().requires_grad_(False)
     missing_keys, extra_keys = autoencoder.load_state_dict(vae_state_dict, strict=False)
-    autoencoer = autoencoder.to(device)
     autoencoder.eval()
     assert len(missing_keys) == 0
 else:

     vae_state_dict = torch.load(vae_path,map_location='cpu')['state_dict']
     autoencoder = vqvae.create_model(3, 3, opts.latent_dim).eval().requires_grad_(False)
     missing_keys, extra_keys = autoencoder.load_state_dict(vae_state_dict, strict=False)
+    autoencoder = autoencoder.to(device)
     autoencoder.eval()
     assert len(missing_keys) == 0
 else: