examples icon indicating copy to clipboard operation
examples copied to clipboard

Add accelerate API support for Word Language Model example

Open framoncg opened this issue 5 months ago • 3 comments

Refactor Word Language Model example to utilize torch.accelerator API torch.accelerator API allows to abstract some of the accelerator specifics in the user scripts. By leveraging this API, the code becomes more adaptable to various hardware accelerators.

Updated word_language_model/main.py with accelerator flag Updated word_language_model/generate.py with accelerator flag Updated README to match word_language_model/main.py flags Updated run_python_examples.sh to add new accelerator flag

CC: @msaroufim, @malfet, @dvrogozh

framoncg avatar May 15 '25 20:05 framoncg