Merge branch 'dev' into test-fp8

This commit is contained in:
Kohaku-Blueleaf
2023-12-02 17:00:09 +08:00
26 changed files with 677 additions and 253 deletions

View File

@@ -55,7 +55,7 @@ def get_optimal_device():
def get_device_for(task):
if task in shared.cmd_opts.use_cpu:
if task in shared.cmd_opts.use_cpu or "all" in shared.cmd_opts.use_cpu:
return cpu
return get_optimal_device()