Added fuyu captioning

2023-10-25 14:14:53 -06:00
parent d742792ee4
commit 9636194c09
10 changed files with 240 additions and 87 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -176,6 +176,7 @@ class StableDiffusion:
                    dtype=dtype,
                    device=self.device_torch,
                    variant="fp16",
+                    use_safetensors=True,
                    **load_args
                )
            else: