diff --git a/otherarch/acestep/dit-vae.cpp b/otherarch/acestep/dit-vae.cpp index b67256228..bffd4f279 100644 --- a/otherarch/acestep/dit-vae.cpp +++ b/otherarch/acestep/dit-vae.cpp @@ -745,6 +745,7 @@ std::string acestep_generate_audio(const music_generation_inputs inputs) req.caption = "An interesting song"; } req.thinking = false; + req.inference_steps = (req.inference_steps>100?100:req.inference_steps); //clamp to 100 const int FRAMES_PER_SECOND = 25; int Oc = music_dit_cfg.out_channels; // 64