Error while using it with transformer

#18
by atifjamilfunsol - opened

I am getting the below error:
Entry Not Found for url: https://huggingface.co/ACE-Step/ACE-Step-v1-3.5B/resolve/main/model_index.json.

My code:

from transformers import AutoModel

# Load the ACE-Step model directly
model = AutoModel.from_pretrained(
    "ACE-Step/Ace-Step1.5",
    trust_remote_code=True,
    dtype="auto"  # automatically chooses float16 or float32 depending on hardware
)

# Example: you would then feed your prompt to the model manually
text_prompt = "A relaxing piano melody with soft strings"

# ACE-Step custom code usually provides a .generate() method for music
audio_tensor = model.generate(text_prompt)

print("Audio generated as a tensor with shape:", audio_tensor.shape)

Sign up or log in to comment