A check script around timing limits

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()