merve HF staff commited on
Commit
69b6e00
1 Parent(s): e5d3156

Upload hiera_mae.py

Browse files
Files changed (1) hide show
  1. hiera/hiera_mae.py +2 -2
hiera/hiera_mae.py CHANGED
@@ -295,7 +295,7 @@ class MaskedAutoencoderHiera(Hiera):
295
  # Image Models
296
 
297
  @pretrained_model({
298
- "mae_in1k": "https://dl.fbaipublicfiles.com/hiera/mae_hiera_tiny_224.pth",
299
  }, default="mae_in1k")
300
  def mae_hiera_tiny_224(**kwargs):
301
  return MaskedAutoencoderHiera(
@@ -304,7 +304,7 @@ def mae_hiera_tiny_224(**kwargs):
304
 
305
 
306
  @pretrained_model({
307
- "mae_in1k": "https://dl.fbaipublicfiles.com/hiera/mae_hiera_small_224.pth",
308
  }, default="mae_in1k")
309
  def mae_hiera_small_224(**kwargs):
310
  return MaskedAutoencoderHiera(
 
295
  # Image Models
296
 
297
  @pretrained_model({
298
+ "mae_in1k": "https://huggingface.co/merve/hiera-tiny-224-in1k/resolve/main/mae_hiera_tiny_224.pth",
299
  }, default="mae_in1k")
300
  def mae_hiera_tiny_224(**kwargs):
301
  return MaskedAutoencoderHiera(
 
304
 
305
 
306
  @pretrained_model({
307
+ "mae_in1k": "https://huggingface.co/merve/hiera-small-224-in1k/resolve/main/mae_hiera_small_224.pth",
308
  }, default="mae_in1k")
309
  def mae_hiera_small_224(**kwargs):
310
  return MaskedAutoencoderHiera(