Upload best checkpoint (epoch 31, val_loss=0.2625)

42d88a9 verified 7 days ago

227 Bytes

	from transformers import AutoModel

	MODEL_ID = "pszmk/mnist-vae-latent2"
	REVISION = "main"

	model = AutoModel.from_pretrained(
	MODEL_ID,
	revision=REVISION,
	trust_remote_code=True,
	)

	print(model.__class__.__name__)