Sync remote code: GRUVAEOutput forward, encoder.encode API

8d93a9a verified 4 months ago

225 Bytes

	from transformers import AutoModel

	MODEL_ID = "pszmk/lamp-grugru-vae"
	REVISION = "main"

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

	print(model.__class__.__name__)