implement args for new backend

This commit is contained in:
layerdiffusion
2024-07-29 11:22:20 -06:00
parent 1b2610db3e
commit 5d291c28ad
2 changed files with 8 additions and 6 deletions
+7
View File
@@ -0,0 +1,7 @@
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--cuda-stream", action="store_true")
args = parser.parse_known_args()[0]