diff --git a/server/scripts/get_limits.py b/server/scripts/get_limits.py new file mode 100644 index 0000000..24a0ed9 --- /dev/null +++ b/server/scripts/get_limits.py @@ -0,0 +1,25 @@ +#!/usr/bin/env python3 +import json +import os +import sys + +ACESTEP_PATH = os.environ.get('ACESTEP_PATH', '/home/ambsd/Desktop/aceui/ACE-Step-1.5') +sys.path.insert(0, ACESTEP_PATH) + +from acestep.gpu_config import get_gpu_config + + +def main(): + cfg = get_gpu_config() + print(json.dumps({ + "tier": cfg.tier, + "gpu_memory_gb": cfg.gpu_memory_gb, + "max_duration_with_lm": cfg.max_duration_with_lm, + "max_duration_without_lm": cfg.max_duration_without_lm, + "max_batch_size_with_lm": cfg.max_batch_size_with_lm, + "max_batch_size_without_lm": cfg.max_batch_size_without_lm, + })) + + +if __name__ == "__main__": + main()