Upload folder using huggingface_hub

myna.py

@@ -20,7 +20,7 @@ import shutil
 
 
 def pair(t):
-    return t if isinstance(t, tuple) else (t, t)
+    return t if isinstance(t, (tuple, list)) else (t, t)
 
 
 def posemb_sincos_2d(h, w, dim, temperature: int = 10000, dtype = torch.float32):
@@ -268,7 +268,7 @@ class Myna(PreTrainedModel, PyTorchModelHubMixin):
 
     def _make_embeddings(self, patch_height, patch_width, patch_dim, dim, image_height, image_width):
         to_patch_embedding = nn.Sequential(
-            Rearrange("b c (h p1) (w p2) -> b (h w) (p1 p2 c)", p1 = patch_height, p2 = patch_width),
+            Rearrange('b c (h p1) (w p2) -> b (h w) (p1 p2 c)', p1 = patch_height, p2 = patch_width),
             nn.LayerNorm(patch_dim),
             nn.Linear(patch_dim, dim),
             nn.LayerNorm(dim),
@@ -286,7 +286,7 @@ class Myna(PreTrainedModel, PyTorchModelHubMixin):
         n_frames = self.config.n_frames
         if n_samples and n_samples != self.config.n_samples:
             n_frames = self.config._get_n_frames(n_samples)
-        spec = self.preprocessor(filename, n_frames)
+        spec = self.preprocessor(filename, n_frames).to(self.device)
         return self(spec)
 
     @property