frozenleaves/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-10-20 17:13:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix typo in Language Modeling example scripts and update TPU type (#38652)

Browse Source 
* Fix typo that prevents the examples to be run correctly

* return .TPU in accelerator.distributedtype comparison
This commit is contained in:
Francisco R Castro Garcia
2025-06-10 07:43:35 -06:00
committed by GitHub
parent 8ff22e9d3b
commit cb4c56ce0d
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/language-modeling/run_fim.py
View File

@ -521,7 +521,7 @@ def main():
# Get the factor by which the embedding layer should be padded based on the device
pad_factor = 1
if torch.cuda.is_availble():
if torch.cuda.is_available():
pad_factor = 8
elif is_torch_xla_available(check_is_tpu=True):

2
examples/pytorch/language-modeling/run_fim_no_trainer.py
View File

@ -488,7 +488,7 @@ def main():
# Get the factor by which the embedding layer should be padded based on the device
pad_factor = 1
if torch.cuda.is_availble():
if torch.cuda.is_available():
pad_factor = 8
elif is_torch_xla_available(check_is_tpu=True):