iloncka
/

vit_tiny_patch16_224_2

Image Classification

Model card Files Files and versions

iloncka commited on Dec 14, 2023

Commit

a6668f9

·

1 Parent(s): c7c93f5

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -5,8 +5,9 @@ tags:
 - image-classification
 datasets:
 - iloncka/mosal
 ---
 Finetuned from the weights from the [timm repository](https://github.com/rwightman/pytorch-image-models). This model is used in the same way as [ViT-base](https://huggingface.co/google/vit-base-patch16-224).
-Note that [safetensors] model requires torch 2.0 environment and fastai.

 - image-classification
 datasets:
 - iloncka/mosal
+library_name: fastai
 ---
 Finetuned from the weights from the [timm repository](https://github.com/rwightman/pytorch-image-models). This model is used in the same way as [ViT-base](https://huggingface.co/google/vit-base-patch16-224).
+Note that [safetensors] model requires torch 2.0 environment and fastai.