Skip to content

Fix missing/broken model weights #14

@a-doering

Description

@a-doering

Please install a copy of your environment to see if broken pytorch_model.bin is the problem or if the environment is the problem. If the environment is the problem, please fix and test with different environment.

27.0.0.1 - - [13/Aug/2023 20:53:05] "GET /static/loadingGif.gif HTTP/1.1" 304 -
768 1024
[2023-08-13 20:53:15,540] ERROR in app: Exception on /generate-image [POST]
Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/transformers/modeling_utils.py", line 463, in load_state_dict
    return torch.load(checkpoint_file, map_location="cpu")
  File "/opt/conda/lib/python3.10/site-packages/torch/serialization.py", line 815, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "/opt/conda/lib/python3.10/site-packages/torch/serialization.py", line 1033, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '"'.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/transformers/modeling_utils.py", line 467, in load_state_dict
    if f.read(7) == "version":
  File "/opt/conda/lib/python3.10/codecs.py", line 322, in decode
    (result, consumed) = self._buffer_decode(data, self.errors, final)
UnicodeDecodeError: 'utf-8' codec can't decode byte 0x8b in position 0: invalid start byte

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/flask/app.py", line 2190, in wsgi_app
    response = self.full_dispatch_request()
  File "/opt/conda/lib/python3.10/site-packages/flask/app.py", line 1486, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "/opt/conda/lib/python3.10/site-packages/flask/app.py", line 1484, in full_dispatch_request
    rv = self.dispatch_request()
  File "/opt/conda/lib/python3.10/site-packages/flask/app.py", line 1469, in dispatch_request
    return self.ensure_sync(self.view_functions[rule.endpoint])(**view_args)
  File "/root/code/ECommerce_Model/./webApp_Flask/webApp_flask.py", line 78, in generate_image_api
    generated_image = generate_image(desired_prompt, segment_prompt, input_image)
  File "/root/code/ECommerce_Model/./webApp_Flask/webApp_flask.py", line 48, in generate_image
    generated_img = inpainting_api(prompt=desired_prompt,
  File "/root/code/ECommerce_Model/modules/api_calls.py", line 54, in inpainting_api
    pipeline = build_SD_pipeline_based_on_input(checkpoint_directory_SD, device, pipeline_type="Inpaint",
  File "/root/code/ECommerce_Model/utils/pipeline_loader.py", line 34, in build_SD_pipeline_based_on_input
    pipe = StableDiffusionInpaintPipeline.from_pretrained(
  File "/opt/conda/lib/python3.10/site-packages/diffusers/pipelines/pipeline_utils.py", line 1037, in from_pretrained
    loaded_sub_model = load_sub_model(
  File "/opt/conda/lib/python3.10/site-packages/diffusers/pipelines/pipeline_utils.py", line 450, in load_sub_model
    loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs)
  File "/opt/conda/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2604, in from_pretrained
    state_dict = load_state_dict(resolved_archive_file)
  File "/opt/conda/lib/python3.10/site-packages/transformers/modeling_utils.py", line 479, in load_state_dict
    raise OSError(
OSError: Unable to load weights from pytorch checkpoint file for './checkpoints/Photon_inpaint/safety_checker/pytorch_model.bin' at './checkpoints/Photon_inpaint/safety_checker/pytorch_model.bin'. If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.
127.0.0.1 - - [13/Aug/2023 20:53:15] "POST /generate-image HTTP/1.1" 500 -

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions