riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix dead flax links modeling_flax_pytorch_utils.py (#38212)

Browse Source
This commit is contained in:
James Niken 2025-05-20 15:03:41 +02:00 committed by GitHub
parent bb3c6426d8
commit 856f034f45
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/modeling_flax_pytorch_utils.py
View File

@ -66,7 +66,7 @@ def load_pytorch_checkpoint_in_flax_state_dict(
except (ImportError, ModuleNotFoundError): except (ImportError, ModuleNotFoundError):
logger.error( logger.error(
"Loading a PyTorch model in Flax, requires both PyTorch and Flax to be installed. Please see" "Loading a PyTorch model in Flax, requires both PyTorch and Flax to be installed. Please see"
" https://pytorch.org/ and https://flax.readthedocs.io/en/latest/installation.html for installation" " https://pytorch.org/ and https://flax.readthedocs.io/en/latest/index.html#installation for installation"
" instructions." " instructions."
) )
raise raise
@ -360,7 +360,7 @@ def load_flax_weights_in_pytorch_model(pt_model, flax_state):
except (ImportError, ModuleNotFoundError): except (ImportError, ModuleNotFoundError):
logger.error( logger.error(
"Loading a Flax weights in PyTorch, requires both PyTorch and Flax to be installed. Please see" "Loading a Flax weights in PyTorch, requires both PyTorch and Flax to be installed. Please see"
" https://pytorch.org/ and https://flax.readthedocs.io/en/latest/installation.html for installation" " https://pytorch.org/ and https://flax.readthedocs.io/en/latest/index.html#installation for installation"
" instructions." " instructions."
) )
raise raise