shravvvv
/

SAG-ViT

Image Classification

image_classification

computer_vision

Model card Files Files and versions Community

SAG-ViT / test_model_loading.py

shravvvv's picture

Updated files

a041fbe 4 months ago

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)