Where to put CompVis/stable-diffusion-safety-checker?

#42
by estherlee - opened

I cannot assess to huggingface.co in the cloud end, so I download in my own computer. But I don't know where to put it.
The reason I need it is because this error:
The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/sd_to_diff.py", line 171, in extract_checkpoint
pipe = StableDiffusionPipeline.from_single_file(
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/venv/lib/python3.9/site-packages/diffusers/loaders.py", line 2308, in from_single_file
pipe = download_from_original_stable_diffusion_ckpt(
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/venv/lib/python3.9/site-packages/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py", line 1562, in download_from_original_stable_diffusion_ckpt
safety_checker = StableDiffusionSafetyChecker.from_pretrained(
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/modeling_utils.py", line 2449, in from_pretrained
config, model_kwargs = cls.config_class.from_pretrained(
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/configuration_utils.py", line 591, in from_pretrained
config_dict, kwargs = cls.get_config_dict(pretrained_model_name_or_path, **kwargs)
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/configuration_utils.py", line 620, in get_config_dict
config_dict, kwargs = cls._get_config_dict(pretrained_model_name_or_path, **kwargs)
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/configuration_utils.py", line 675, in _get_config_dict
resolved_config_file = cached_file(
File "/mnt/workspace/base/stablediffusion/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/utils/hub.py", line 468, in cached_file
raise EnvironmentError(
OSError: We couldn't connect to 'https://huggingface.co' to load this file, couldn't find it in the cached files and it looks like CompVis/stable-diffusion-safety-checker is not the path to a directory containing a file named config.json.
Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/transformers/installation#offline-mode'.
Couldn't find /mnt/workspace/base/stablediffusion/stable-diffusion-webui/models/dreambooth/cloth/working/unet
Unable to extract checkpoint!
======================

I was using dreambooth to create a model of my own!

dir:.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker
local down:

model = AutoFeatureExtractor.from_pretrained('CompVis/stable-diffusion-safety-checker')
safety_checker = StableDiffusionSafetyChecker.from_pretrained('CompVis/stable-diffusion-safety-checker')

Sign up or log in to comment