Improvements to dataloader

2026-04-24 08:19:24 +00:00 · 2024-04-27 09:28:28 -06:00
parent 5da3613e0b
commit b96913d73c
5 changed files with 21 additions and 17 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -330,6 +330,7 @@ class StableDiffusion:
                    requires_safety_checker=False,
                    safety_checker=None,
                    variant="fp16",
+                    trust_remote_code=True,
                    **load_args
                ).to(self.device_torch)
            else:
@@ -341,6 +342,7 @@ class StableDiffusion:
                    requires_safety_checker=False,
                    torch_dtype=self.torch_dtype,
                    safety_checker=None,
+                    trust_remote_code=True,
                    **load_args
                ).to(self.device_torch)
            flush()