huggingface_main() — pytorch Function Reference

Architecture documentation for the huggingface_main() function in huggingface.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration calls 2 called by 1

Entity Profile

DynamoBenchmarks→ Configuration→ huggingface_main() — pytorch Function Reference

Dependency Diagram

graph TD
  b93ae0ef_d85a_81fb_b07f_38119c7d07cc["huggingface_main()"]
  815a46f9_d548_98bf_cbd7_cb9fae326824["args()"]
  815a46f9_d548_98bf_cbd7_cb9fae326824 -->|calls| b93ae0ef_d85a_81fb_b07f_38119c7d07cc
  11c86f38_3751_f253_e544_6933536508cd["refresh_model_names_and_batch_sizes()"]
  b93ae0ef_d85a_81fb_b07f_38119c7d07cc -->|calls| 11c86f38_3751_f253_e544_6933536508cd
  60550900_f047_a85b_c5ab_ff80b30486b4["main()"]
  b93ae0ef_d85a_81fb_b07f_38119c7d07cc -->|calls| 60550900_f047_a85b_c5ab_ff80b30486b4
  style b93ae0ef_d85a_81fb_b07f_38119c7d07cc fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/huggingface.py lines 656–662

def huggingface_main():
    # Code to refresh model names and batch sizes
    # if "--find-batch-sizes" not in sys.argv:
    #     refresh_model_names_and_batch_sizes()
    logging.basicConfig(level=logging.WARNING)
    warnings.filterwarnings("ignore")
    main(HuggingfaceRunner())