Merge branch 'pin-options-panel' of https://github.com/psychedelicious/stable-diffusion into psychedelicious-pin-options-panel

- from PR #1301
2024-08-30 20:32:17 +00:00 · 2022-10-31 09:37:13 -04:00
parent e8aba99c92 9a15a89e20
commit 89da42ad79
114 changed files with 2034 additions and 1232 deletions
--- a/ldm/generate.py
+++ b/ldm/generate.py
@ -219,6 +219,7 @@ class Generate:
                safety_model_id = "CompVis/stable-diffusion-safety-checker"
                self.safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id, local_files_only=True)
                self.safety_feature_extractor = AutoFeatureExtractor.from_pretrained(safety_model_id, local_files_only=True)
+                self.safety_checker.to(self.device)
            except Exception:
                print('** An error was encountered while installing the safety checker:')
                print(traceback.format_exc())