Cloyir/index-tts2-ForDgxSpark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(cli): Support XPU (#322)

Browse Source 
* Support XPU

* Support XPU
This commit is contained in:
DDXDB 2025-09-10 22:35:06 +08:00 committed by GitHub
parent 242604d27e
commit e83df4e427
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
indextts/cli.py
View File

@ -14,7 +14,7 @@ def main():
parser.add_argument("--model_dir", type=str, default="checkpoints", help="Path to the model directory. Default is 'checkpoints'")
parser.add_argument("--fp16", action="store_true", default=False, help="Use FP16 for inference if available")
parser.add_argument("-f", "--force", action="store_true", default=False, help="Force to overwrite the output file if it exists")
parser.add_argument("-d", "--device", type=str, default=None, help="Device to run the model on (cpu, cuda, mps)." )
parser.add_argument("-d", "--device", type=str, default=None, help="Device to run the model on (cpu, cuda, mps, xpu)." )
args = parser.parse_args()
if len(args.text.strip()) == 0:
print("ERROR: Text is empty.")
@ -49,6 +49,8 @@ def main():
args.device = "cuda:0"
elif torch.mps.is_available():
args.device = "mps"
elif torch.xpu.is_available():
args.device = "xpu"
else:
args.device = "cpu"
args.fp16 = False # Disable FP16 on CPU