How to use mohammadmahdinouri/interleaved-init-base with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, FlowModelForDistillation tokenizer = AutoTokenizer.from_pretrained("mohammadmahdinouri/interleaved-init-base") model = FlowModelForDistillation.from_pretrained("mohammadmahdinouri/interleaved-init-base", device_map="auto")
