Merge branch 'dev' into test-fp8

This commit is contained in:
Kohaku-Blueleaf 2023-12-02 17:00:09 +08:00
commit 110485d5bb
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()