feat: Device override cli arg
This commit is contained in:
parent
97e84a97db
commit
e81997f129
2 changed files with 2 additions and 1 deletions
2
main.py
2
main.py
|
|
@ -7,7 +7,7 @@ from src.utils import determine_device
|
|||
def main():
|
||||
args, print_help = parse_arguments()
|
||||
|
||||
device = determine_device()
|
||||
device = args.device or determine_device()
|
||||
print(f"Running on device: {device}...")
|
||||
|
||||
match args.mode:
|
||||
|
|
|
|||
Reference in a new issue