Duplicated from ACE-Step/acestep-5Hz-lm-4B
How to use diskrot/acestep-5Hz-lm-4B-diskrot with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-to-audio", model="diskrot/acestep-5Hz-lm-4B-diskrot")
# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("diskrot/acestep-5Hz-lm-4B-diskrot") model = AutoModelForCausalLM.from_pretrained("diskrot/acestep-5Hz-lm-4B-diskrot", device_map="auto")
