Fix

2023-06-04 12:42:22 -05:00
parent 392355ea82
commit 44e342eaa1
1 changed files with 1 additions and 1 deletions
@@ -141,7 +141,7 @@ class GuidedDiffusionLoader:
    def __init__(self):
        pass

-    def load(self, model_name, use_checkpoint, use_secondary):
+    def load(self, model_name, use_checkpoint, use_secondary=False):
        use_cpu = args.cpu
        
        use_checkpoint = True if use_checkpoint == "True" else False