1from trainer.scratch_model import ScratchModelCreator
2
3creator = ScratchModelCreator()
4# Load from local
5model, tokenizer, config = creator.load_with_tokenizer("path/to/model")
6
7# Or load from HuggingFace Hub
8local_path = creator.download_from_hub("username/scratch-model-name")
9model, tokenizer, config = creator.load_with_tokenizer(local_path)ScratchTransformer class to load. Use the ScratchModelCreator as shown above.