Skip to content

Commit

Permalink
Merge branch 'ko3n1g/fix/mcore-version' into 'main'
Browse files Browse the repository at this point in the history
fix: Expose __version__ and other package metadata as package attributes

See merge request ADLR/megatron-lm!1521
  • Loading branch information
terrykong committed Jun 16, 2024
2 parents 195496b + e4e5dd6 commit 13c762e
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions megatron/core/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,19 @@
from megatron.core.distributed import DistributedDataParallel
from megatron.core.inference_params import InferenceParams
from megatron.core.model_parallel_config import ModelParallelConfig
from megatron.core.package_info import (
__contact_emails__,
__contact_names__,
__description__,
__download_url__,
__homepage__,
__keywords__,
__license__,
__package_name__,
__repository_url__,
__shortversion__,
__version__,
)
from megatron.core.timers import Timers

# Alias parallel_state as mpu, its legacy name
Expand Down

0 comments on commit 13c762e

Please sign in to comment.