Added fuyu captioning

2023-10-25 14:14:53 -06:00
1 parent d742792ee4
commit 9636194c09
@@ -176,6 +176,7 @@ class StableDiffusion:
                    dtype=dtype,
                    device=self.device_torch,
                    variant="fp16",
+                    use_safetensors=True,
                    **load_args
                )
            else: