Upload model

hf_model.py

@@ -129,8 +129,6 @@ class ERADIOModel(PreTrainedModel):
 
         if isinstance(y, tuple):
             summary, features = y
-            # ERADIO features are spatial tokens.
-            features = rearrange(features, 'b c h w -> b (h w) c')
         else:
             summary = y
             features = None