Intel
/

tiny-random-vit_ipex_model

Image Classification

Inference Endpoints

Model card Files Files and versions Community

Jiqing commited on 23 days ago

Commit

e6aef6e

•

1 Parent(s): 923c185

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -5,12 +5,12 @@ tags: []
 # Model Card for Model ID
-This is a tiny random vit model derived from "google/vit-base-patch16-224". It was uploaded by IPEXModelForCausalLM.
 ```
-from optimum.intel import IPEXModelForCausalLM
-model = IPEXModelForCausalLM.from_pretrained("hf-internal-testing/tiny-random-vit")
 model.push_to_hub("Intel/tiny-random-vit_ipex_model")
 ```

 # Model Card for Model ID
+This is a tiny random vit model derived from "google/vit-base-patch16-224". It was uploaded by IPEXModelForImageClassification.
 ```
+from optimum.intel import IPEXModelForImageClassification
+model = IPEXModelForImageClassification.from_pretrained("hf-internal-testing/tiny-random-vit")
 model.push_to_hub("Intel/tiny-random-vit_ipex_model")
 ```