Skip to content

GitLab

Unverified Commit ec64f371 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Chore] remove tf mention (#7245) 

remove tf mention
parent cd6e1f11
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
src/diffusers/models/modeling_utils.py
View file @ ec64f371
...@@ -124,9 +124,7 @@ def load_state_dict(checkpoint_file: Union[str, os.PathLike], variant: Optional[ ...@@ -124,9 +124,7 @@ def load_state_dict(checkpoint_file: Union[str, os.PathLike], variant: Optional[
) from e ) from e
except (UnicodeDecodeError, ValueError): except (UnicodeDecodeError, ValueError):
raise OSError( raise OSError(
f"Unable to load weights from checkpoint file for '{checkpoint_file}' " f"Unable to load weights from checkpoint file for '{checkpoint_file}' " f"at '{checkpoint_file}'. "
f"at '{checkpoint_file}'. "
"If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True."
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment