shravvvv
/

SAG-ViT

Image Classification

image_classification

computer_vision

Model card Files Files and versions

SAG-ViT / test_model_loading.py

shravvvv's picture

Updated files

a041fbe almost 2 years ago

Raw History Blame Contribute Delete

225 Bytes

	from transformers import AutoModel, AutoConfig

	# Load Config and Model
	config = AutoConfig.from_pretrained("shravvvv/SAG-ViT")
	model = AutoModel.from_pretrained("shravvvv/SAG-ViT", config=config)

	# Check Model
	print(model)