Spaces:

huzey
/

ncut-pytorch

Running on Zero

huzey commited on Aug 29

Commit

59742d8

•

1 Parent(s): 9e43ce3

update UI

Files changed (1) hide show

backbone.py CHANGED Viewed

@@ -657,9 +657,9 @@ class EVA02(nn.Module):
             'block': block_outputs
         }
-MODEL_DICT["CLIP(eva02_base_patch14_448.mim_in22k_ft_in1k)"] = partial(EVA02)
-LAYER_DICT["CLIP(eva02_base_patch14_448.mim_in22k_ft_in1k)"] = 12
-RES_DICT["CLIP(eva02_base_patch14_448.mim_in22k_ft_in1k)"] = (448, 448)
 class CLIPConvnext(nn.Module):
     def __init__(self):

             'block': block_outputs
         }
+MODEL_DICT["CLIP(eva02_base_patch14_448/mim_in22k_ft_in1k)"] = partial(EVA02)
+LAYER_DICT["CLIP(eva02_base_patch14_448/mim_in22k_ft_in1k)"] = 12
+RES_DICT["CLIP(eva02_base_patch14_448/mim_in22k_ft_in1k)"] = (448, 448)
 class CLIPConvnext(nn.Module):
     def __init__(self):