Spaces:

mantrakp
/

aai

Runtime error

barreloflube commited on Sep 16, 2024

Commit

a5ebb54

1 Parent(s): 12289cb

Refactor model loading in load_models.py to use ControlNetModel instead of FluxControlNetModel

Files changed (1) hide show

tabs/images/load_models.py CHANGED Viewed

@@ -44,7 +44,7 @@ def init_sys():
                 torch_dtype=controlnet['compute_type']
             ).to(device)])
         elif controlnet['loader'] == 'sdxl':
-            controlnet['controlnet'] = FluxControlNetModel.from_pretrained(
                 controlnet['repo_id'],
                 torch_dtype=controlnet['compute_type']
             ).to(device)

                 torch_dtype=controlnet['compute_type']
             ).to(device)])
         elif controlnet['loader'] == 'sdxl':
+            controlnet['controlnet'] = ControlNetModel.from_pretrained(
                 controlnet['repo_id'],
                 torch_dtype=controlnet['compute_type']
             ).to(device)