fix(critical/BREAKING): correct config, regenerate weights, add ONNX v1.1, requires transformers >= 5.4.0 (#5)

0b8f9cb 4 days ago

524 Bytes

	{
	"architectures": [
	"ViTForImageClassification"
	],
	"attention_probs_dropout_prob": 0.0,
	"hidden_act": "gelu",
	"hidden_dropout_prob": 0.0,
	"hidden_size": 384,
	"image_size": 384,
	"initializer_range": 0.02,
	"input_size": 384,
	"intermediate_size": 1536,
	"layer_norm_eps": 1e-06,
	"model_type": "vit",
	"num_attention_heads": 6,
	"num_channels": 3,
	"num_labels": 1,
	"num_hidden_layers": 12,
	"patch_size": 16,
	"qkv_bias": true,
	"torch_dtype": "float32",
	"transformers_version": "5.4.0"
	}