Format code (#624)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2023-06-24 18:06:17 +08:00
committed by GitHub
parent 04d7813486
commit fad31f24f5
4 changed files with 11 additions and 5 deletions

View File

@@ -1,6 +1,6 @@
import os, sys, traceback
os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
# device=sys.argv[1]
n_part = int(sys.argv[2])