当前位置:   article > 正文

运行StableDiffusionInpaintPipeline的Example时报错:OSError: Cannot load model runwayml/stable-diffusion-..._oserror: cannot load model v1.5: model is not cach

oserror: cannot load model v1.5: model is not cached locally and an error oc

项目地址:

https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/inpainticon-default.png?t=N7T8https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/inpaint在云服务器端运行下面给出的Example的时候出现报错:

  1. (myconda) root@wnyrpE:/mnt/Diffusers/inpainting# python3 SD_inpaint_eg.py
  2. Couldn't connect to the Hub: (MaxRetryError("HTTPSConnectionPool(host='huggingface.co', port=443): Max retries exceeded with url: /api/models/runwayml/stable-diffusion-inpainting (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x7fa6c5357dc0>: Failed to establish a new connection: [Errno 110] Connection timed out'))"), '(Request ID: bebba422-c4d8-4797-a818-a00fe02957e8)').
  3. Will try to load from local cache.
  4. Traceback (most recent call last):
  5. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/connection.py", line 174, in _new_conn
  6. conn = connection.create_connection(
  7. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/util/connection.py", line 95, in create_connection
  8. raise err
  9. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/util/connection.py", line 85, in create_connection
  10. sock.connect(sa)
  11. TimeoutError: [Errno 110] Connection timed out
  12. During handling of the above exception, another exception occurred:
  13. Traceback (most recent call last):
  14. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/connectionpool.py", line 703, in urlopen
  15. httplib_response = self._make_request(
  16. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/connectionpool.py", line 386, in _make_request
  17. self._validate_conn(conn)
  18. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/connectionpool.py", line 1040, in _validate_conn
  19. conn.connect()
  20. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/connection.py", line 358, in connect
  21. self.sock = conn = self._new_conn()
  22. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/connection.py", line 186, in _new_conn
  23. raise NewConnectionError(
  24. urllib3.exceptions.NewConnectionError: <urllib3.connection.HTTPSConnection object at 0x7fa6c5357dc0>: Failed to establish a new connection: [Errno 110] Connection timed out
  25. During handling of the above exception, another exception occurred:
  26. Traceback (most recent call last):
  27. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/requests/adapters.py", line 440, in send
  28. resp = conn.urlopen(
  29. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/connectionpool.py", line 785, in urlopen
  30. retries = retries.increment(
  31. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/urllib3/util/retry.py", line 592, in increment
  32. raise MaxRetryError(_pool, url, error or ResponseError(cause))
  33. urllib3.exceptions.MaxRetryError: HTTPSConnectionPool(host='huggingface.co', port=443): Max retries exceeded with url: /api/models/runwayml/stable-diffusion-inpainting (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x7fa6c5357dc0>: Failed to establish a new connection: [Errno 110] Connection timed out'))
  34. During handling of the above exception, another exception occurred:
  35. Traceback (most recent call last):
  36. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/diffusers/pipelines/pipeline_utils.py", line 1656, in download
  37. info = model_info(pretrained_model_name, token=token, revision=revision)
  38. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/huggingface_hub/utils/_validators.py", line 118, in _inner_fn
  39. return fn(*args, **kwargs)
  40. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/huggingface_hub/hf_api.py", line 2084, in model_info
  41. r = get_session().get(path, headers=headers, timeout=timeout, params=params)
  42. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/requests/sessions.py", line 542, in get
  43. return self.request('GET', url, **kwargs)
  44. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/requests/sessions.py", line 529, in request
  45. resp = self.send(prep, **send_kwargs)
  46. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/requests/sessions.py", line 645, in send
  47. r = adapter.send(request, **kwargs)
  48. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/huggingface_hub/utils/_http.py", line 67, in send
  49. return super().send(request, *args, **kwargs)
  50. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/requests/adapters.py", line 519, in send
  51. raise ConnectionError(e, request=request)
  52. requests.exceptions.ConnectionError: (MaxRetryError("HTTPSConnectionPool(host='huggingface.co', port=443): Max retries exceeded with url: /api/models/runwayml/stable-diffusion-inpainting (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x7fa6c5357dc0>: Failed to establish a new connection: [Errno 110] Connection timed out'))"), '(Request ID: bebba422-c4d8-4797-a818-a00fe02957e8)')
  53. The above exception was the direct cause of the following exception:
  54. Traceback (most recent call last):
  55. File "SD_inpaint_eg.py", line 20, in <module>
  56. pipe = StableDiffusionInpaintPipeline.from_pretrained(
  57. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/huggingface_hub/utils/_validators.py", line 118, in _inner_fn
  58. return fn(*args, **kwargs)
  59. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/diffusers/pipelines/pipeline_utils.py", line 1096, in from_pretrained
  60. cached_folder = cls.download(
  61. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/huggingface_hub/utils/_validators.py", line 118, in _inner_fn
  62. return fn(*args, **kwargs)
  63. File "/root/miniconda3/envs/myconda/lib/python3.8/site-packages/diffusers/pipelines/pipeline_utils.py", line 1905, in download
  64. raise EnvironmentError(
  65. OSError: Cannot load model runwayml/stable-diffusion-inpainting: model is not cached locally and an error occured while trying to fetch metadata from the Hub. Please check out the root cause in the stacktrace above.
原因是:国内无法服务器无法直接连接上huggingface。

解决办法是:开代理把模型下载到本地再上传到服务器端。(不管是下载到本地,还是直接同步到服务器端,只要是国内的网络都要把代理打开)

我运行的Example代码是:

  1. import PIL
  2. import requests
  3. import torch
  4. from io import BytesIO
  5. from diffusers import StableDiffusionInpaintPipeline
  6. def download_image(url):
  7. response = requests.get(url)
  8. return PIL.Image.open(BytesIO(response.content)).convert("RGB")
  9. img_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
  10. mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
  11. init_image = download_image(img_url).resize((512, 512))
  12. mask_image = download_image(mask_url).resize((512, 512))
  13. pipe = StableDiffusionInpaintPipeline.from_pretrained(
  14. "runwayml/stable-diffusion-inpainting", torch_dtype=torch.float16
  15. )
  16. pipe = pipe.to("cuda")
  17. prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
  18. image = pipe(prompt=prompt, image=init_image, mask_image=mask_image).images[0]

从Example代码里面看到,应该是要下载runwayml/stable-diffusion-inpainting模型

  1. pipe = StableDiffusionInpaintPipeline.from_pretrained(
  2. "runwayml/stable-diffusion-inpainting", torch_dtype=torch.float16
  3. )
  4. pipe = pipe.to("cuda")
注意修改脚本中的代码:
  1. pipe = StableDiffusionInpaintPipeline.from_pretrained(
  2. "/本地/模型/路径", torch_dtype=torch.float16
  3. )
  4. pipe = pipe.to("cuda")

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/804235
推荐阅读
相关标签
  

闽ICP备14008679号