PulkundwarP
/

Try2

Try2 / __init__.py

Upload 4 files

93e6aa5 verified about 1 year ago

188 Bytes

	from .configuration_custom_gpt import CustomGPTConfig
	from .custom_gpt import CustomGPT

	CustomGPTConfig.register_for_auto_class()
	CustomGPT.register_for_auto_class("AutoModelForCausalLM")