riversedge
26 lines
659 B
Python
26 lines
659 B
Python
#!/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()
|