Use backend enum on diffusers check.

This commit is contained in:
Seunghoon Lee
2023-07-06 01:45:12 +09:00
parent 993de932ab
commit a076ff1b43
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -1,5 +1,5 @@
from modules import shared
def override_opts():
if shared.cmd_opts.backend.lower() == "diffusers":
if shared.backend == shared.Backend.DIFFUSERS:
shared.opts.diffusers_generator_device = "cpu" # DirectML does not support torch.Generator API.