Safetensors
vit

lat_mean=39.951163 lat_std=0.000596 lon_mean=-75.19156 lon_std=0.000609

class CustomViTModel(ViTForImageClassification, PyTorchModelHubMixin): def init(self, config): super().init(config) self.classifier = nn.Linear(self.classifier.in_features, config.num_labels)

model = CustomViTModel.from_pretrained("img2gpspenn/vit") model.eval()

Downloads last month
25
Safetensors
Model size
5.52M params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.