lightning 2.5.2__tar.gz → 2.5.3__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {lightning-2.5.2/src/lightning.egg-info → lightning-2.5.3}/PKG-INFO +60 -59
- {lightning-2.5.2 → lightning-2.5.3}/README.md +6 -0
- {lightning-2.5.2 → lightning-2.5.3}/pyproject.toml +1 -0
- {lightning-2.5.2 → lightning-2.5.3}/requirements/base.txt +3 -3
- {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/base.txt +4 -4
- {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/examples.txt +2 -3
- {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/strategies.txt +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/test.txt +3 -3
- {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/base.txt +6 -6
- {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/docs.txt +3 -1
- {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/examples.txt +2 -3
- {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/extra.txt +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/strategies.txt +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/requirements/pytorch/test.txt +5 -4
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__about__.py +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/CHANGELOG.md +13 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/registry.py +2 -2
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/cli.py +10 -2
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/xla.py +15 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/bitsandbytes.py +5 -3
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/consolidate_checkpoint.py +6 -5
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/device_parser.py +15 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/imports.py +2 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/seed.py +4 -5
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/spike.py +1 -4
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/testing/_runif.py +7 -5
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/CHANGELOG.md +24 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/lr_finder.py +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/model_checkpoint.py +49 -6
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/rich_progress.py +39 -5
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/module.py +62 -11
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/optimizer.py +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/transformer.py +13 -4
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/loop.py +8 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/manual.py +2 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/training_epoch_loop.py +5 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/advanced.py +6 -4
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/call.py +2 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +6 -11
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/data_connector.py +8 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/lr_finder.py +12 -25
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/imports.py +4 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_summary/model_summary.py +17 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/parsing.py +2 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/testing/_runif.py +11 -4
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +3 -2
- lightning-2.5.3/src/lightning/version.info +1 -0
- {lightning-2.5.2 → lightning-2.5.3/src/lightning.egg-info}/PKG-INFO +60 -59
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/SOURCES.txt +1 -1
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/requires.txt +34 -39
- lightning-2.5.3/src/version.info +1 -0
- lightning-2.5.2/src/lightning/version.info +0 -1
- lightning-2.5.2/src/version.info +0 -1
- {lightning-2.5.2 → lightning-2.5.3}/.actions/assistant.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/CITATION.cff +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/LICENSE +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/requirements/fabric/docs.txt +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/setup.cfg +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/setup.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__setup__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/__version__.py +0 -0
- /lightning-2.5.2/src/lightning/ffd6261c9058703d8c54093b4f93e2b4008030713613e504 → /lightning-2.5.3/src/lightning/b7f7dd8e64cd613ff6563bfe1965950a2c3243b7be6e14de +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/data/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/_graveyard/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/_graveyard/tpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/accelerator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/cpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/cuda.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/mps.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/accelerators/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/fabric.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/csv_logs.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/logger.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/loggers/tensorboard.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/single_device.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/collectives/torch_collective.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/lightning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/lsf.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/torch_io.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/io/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/amp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/double.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/half.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/precision.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/transformer_engine.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/utils.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/plugins/precision/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/ddp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/dp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/multiprocessing.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/subprocess_script.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/model_parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/registry.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/single_device.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/single_xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/strategy.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/strategies/xla_fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/apply_func.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/cloud_io.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/data.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/device_dtype_mixin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/distributed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/enums.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/exceptions.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/init.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/load.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/logger.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/optimizer.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/rank_zero.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/registry.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/throughput.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/types.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/utilities/warnings.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/fabric/wrappers.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/py.typed +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/precision.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/cpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/cuda.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/mps.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/accelerators/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/callback.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/device_stats_monitor.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/early_stopping.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/finetuning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/lr_monitor.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/model_summary.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/prediction_writer.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/progress_bar.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/pruning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/rich_model_summary.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/spike.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/throughput_monitor.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/callbacks/timer.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/cli.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/datamodule.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/hooks.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/mixins/hparams_mixin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/core/saving.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/boring_classes.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/lstm.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/comet.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/csv_logs.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/logger.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/mlflow.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/neptune.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/tensorboard.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/utilities.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loggers/wandb.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/evaluation_loop.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/fetchers.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/fit_loop.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/automatic.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/prediction_loop.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/progress.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/loops/utilities.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/overrides/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/overrides/distributed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/async_plugin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/wrapper.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/amp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/double.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/half.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/precision.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/base.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/profiler.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/pytorch.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/simple.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/profilers/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/serve/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/serve/servable_module.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/serve/servable_module_validator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/ddp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/model_parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/single_device.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/single_xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/strategy.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/strategies/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/callback_connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/connectors/signal_connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/setup.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/states.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/trainer/trainer.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/batch_size_scaling.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/tuner/tuning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/_pytree.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/argparse.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/combined_loader.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/compile.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/data.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/enums.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/exceptions.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/grads.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/memory.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/migration/utils.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_helpers.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_registry.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/seed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/types.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning/pytorch/utilities/warnings.py +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/dependency_links.txt +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/entry_points.txt +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/not-zip-safe +0 -0
- {lightning-2.5.2 → lightning-2.5.3}/src/lightning.egg-info/top_level.txt +0 -0
|
@@ -1,11 +1,11 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: lightning
|
|
3
|
-
Version: 2.5.
|
|
3
|
+
Version: 2.5.3
|
|
4
4
|
Summary: The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
|
|
5
5
|
Home-page: https://github.com/Lightning-AI/lightning
|
|
6
6
|
Download-URL: https://github.com/Lightning-AI/lightning
|
|
7
7
|
Author: Lightning AI et al.
|
|
8
|
-
Author-email:
|
|
8
|
+
Author-email: developer@lightning.ai
|
|
9
9
|
License: Apache-2.0
|
|
10
10
|
Project-URL: Bug Tracker, https://github.com/Lightning-AI/pytorch-lightning/issues
|
|
11
11
|
Project-URL: Documentation, https://lightning.ai/lightning-docs
|
|
@@ -27,93 +27,89 @@ Classifier: Programming Language :: Python :: 3.12
|
|
|
27
27
|
Requires-Python: >=3.9
|
|
28
28
|
Description-Content-Type: text/markdown
|
|
29
29
|
License-File: LICENSE
|
|
30
|
-
Requires-Dist: PyYAML<8.0
|
|
30
|
+
Requires-Dist: PyYAML<8.0,>5.4
|
|
31
31
|
Requires-Dist: fsspec[http]<2027.0,>=2022.5.0
|
|
32
32
|
Requires-Dist: lightning-utilities<2.0,>=0.10.0
|
|
33
33
|
Requires-Dist: packaging<27.0,>=20.0
|
|
34
34
|
Requires-Dist: torch<4.0,>=2.1.0
|
|
35
|
-
Requires-Dist: torchmetrics<3.0
|
|
35
|
+
Requires-Dist: torchmetrics<3.0,>0.7.0
|
|
36
36
|
Requires-Dist: tqdm<6.0,>=4.57.0
|
|
37
|
-
Requires-Dist: typing-extensions<6.0
|
|
37
|
+
Requires-Dist: typing-extensions<6.0,>4.5.0
|
|
38
38
|
Requires-Dist: pytorch-lightning
|
|
39
|
+
Provides-Extra: pytorch-extra
|
|
40
|
+
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
|
|
41
|
+
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-extra"
|
|
42
|
+
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-extra"
|
|
43
|
+
Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-extra"
|
|
44
|
+
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
|
|
45
|
+
Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
|
|
46
|
+
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
|
|
39
47
|
Provides-Extra: pytorch-strategies
|
|
40
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
48
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
|
|
41
49
|
Provides-Extra: pytorch-test
|
|
42
50
|
Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-test"
|
|
43
|
-
Requires-Dist: coverage==7.
|
|
51
|
+
Requires-Dist: coverage==7.10.3; extra == "pytorch-test"
|
|
44
52
|
Requires-Dist: fastapi; extra == "pytorch-test"
|
|
45
53
|
Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-test"
|
|
46
54
|
Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-test"
|
|
47
55
|
Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-test"
|
|
56
|
+
Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "pytorch-test"
|
|
48
57
|
Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-test"
|
|
49
58
|
Requires-Dist: psutil<8.0; extra == "pytorch-test"
|
|
50
59
|
Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-test"
|
|
51
|
-
Requires-Dist: pytest-random-order==1.
|
|
60
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-test"
|
|
52
61
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-test"
|
|
53
62
|
Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-test"
|
|
54
|
-
Requires-Dist: pytest==8.4.
|
|
63
|
+
Requires-Dist: pytest==8.4.1; extra == "pytorch-test"
|
|
55
64
|
Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-test"
|
|
56
65
|
Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "pytorch-test"
|
|
57
66
|
Requires-Dist: uvicorn; extra == "pytorch-test"
|
|
58
|
-
Provides-Extra: pytorch-extra
|
|
59
|
-
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
|
|
60
|
-
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-extra"
|
|
61
|
-
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-extra"
|
|
62
|
-
Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-extra"
|
|
63
|
-
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
|
|
64
|
-
Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
|
|
65
|
-
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
|
|
66
67
|
Provides-Extra: pytorch-examples
|
|
67
68
|
Requires-Dist: ipython[all]<9.0; extra == "pytorch-examples"
|
|
68
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-examples"
|
|
69
69
|
Requires-Dist: requests<3.0; extra == "pytorch-examples"
|
|
70
70
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-examples"
|
|
71
71
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-examples"
|
|
72
72
|
Provides-Extra: fabric-strategies
|
|
73
73
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-strategies"
|
|
74
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
74
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
|
|
75
75
|
Provides-Extra: fabric-test
|
|
76
76
|
Requires-Dist: click==8.1.8; extra == "fabric-test"
|
|
77
|
-
Requires-Dist: coverage==7.
|
|
77
|
+
Requires-Dist: coverage==7.10.3; extra == "fabric-test"
|
|
78
78
|
Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-test"
|
|
79
79
|
Requires-Dist: pytest-cov==6.2.1; extra == "fabric-test"
|
|
80
|
-
Requires-Dist: pytest-random-order==1.
|
|
80
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-test"
|
|
81
81
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-test"
|
|
82
82
|
Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-test"
|
|
83
|
-
Requires-Dist: pytest==8.4.
|
|
83
|
+
Requires-Dist: pytest==8.4.1; extra == "fabric-test"
|
|
84
84
|
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-test"
|
|
85
85
|
Provides-Extra: fabric-examples
|
|
86
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-examples"
|
|
87
86
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-examples"
|
|
88
87
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-examples"
|
|
89
88
|
Provides-Extra: fabric-all
|
|
90
89
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-all"
|
|
91
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
92
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-all"
|
|
90
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
|
|
93
91
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-all"
|
|
94
92
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-all"
|
|
95
93
|
Provides-Extra: fabric-dev
|
|
96
94
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-dev"
|
|
97
95
|
Requires-Dist: click==8.1.8; extra == "fabric-dev"
|
|
98
|
-
Requires-Dist: coverage==7.
|
|
99
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
100
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-dev"
|
|
96
|
+
Requires-Dist: coverage==7.10.3; extra == "fabric-dev"
|
|
97
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
|
|
101
98
|
Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-dev"
|
|
102
99
|
Requires-Dist: pytest-cov==6.2.1; extra == "fabric-dev"
|
|
103
|
-
Requires-Dist: pytest-random-order==1.
|
|
100
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-dev"
|
|
104
101
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-dev"
|
|
105
102
|
Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-dev"
|
|
106
|
-
Requires-Dist: pytest==8.4.
|
|
103
|
+
Requires-Dist: pytest==8.4.1; extra == "fabric-dev"
|
|
107
104
|
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-dev"
|
|
108
105
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-dev"
|
|
109
106
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-dev"
|
|
110
107
|
Provides-Extra: pytorch-all
|
|
111
108
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-all"
|
|
112
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
109
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
|
|
113
110
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-all"
|
|
114
111
|
Requires-Dist: ipython[all]<9.0; extra == "pytorch-all"
|
|
115
112
|
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-all"
|
|
116
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-all"
|
|
117
113
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-all"
|
|
118
114
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-all"
|
|
119
115
|
Requires-Dist: requests<3.0; extra == "pytorch-all"
|
|
@@ -124,25 +120,25 @@ Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-all"
|
|
|
124
120
|
Provides-Extra: pytorch-dev
|
|
125
121
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-dev"
|
|
126
122
|
Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-dev"
|
|
127
|
-
Requires-Dist: coverage==7.
|
|
128
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
123
|
+
Requires-Dist: coverage==7.10.3; extra == "pytorch-dev"
|
|
124
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
|
|
129
125
|
Requires-Dist: fastapi; extra == "pytorch-dev"
|
|
130
126
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-dev"
|
|
131
127
|
Requires-Dist: ipython[all]<9.0; extra == "pytorch-dev"
|
|
132
128
|
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-dev"
|
|
133
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-dev"
|
|
134
129
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
|
|
135
130
|
Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-dev"
|
|
136
131
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-dev"
|
|
137
132
|
Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-dev"
|
|
138
133
|
Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-dev"
|
|
134
|
+
Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "pytorch-dev"
|
|
139
135
|
Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-dev"
|
|
140
136
|
Requires-Dist: psutil<8.0; extra == "pytorch-dev"
|
|
141
137
|
Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-dev"
|
|
142
|
-
Requires-Dist: pytest-random-order==1.
|
|
138
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-dev"
|
|
143
139
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-dev"
|
|
144
140
|
Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-dev"
|
|
145
|
-
Requires-Dist: pytest==8.4.
|
|
141
|
+
Requires-Dist: pytest==8.4.1; extra == "pytorch-dev"
|
|
146
142
|
Requires-Dist: requests<3.0; extra == "pytorch-dev"
|
|
147
143
|
Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-dev"
|
|
148
144
|
Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-dev"
|
|
@@ -151,39 +147,39 @@ Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-dev"
|
|
|
151
147
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-dev"
|
|
152
148
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-dev"
|
|
153
149
|
Requires-Dist: uvicorn; extra == "pytorch-dev"
|
|
150
|
+
Provides-Extra: extra
|
|
151
|
+
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
|
|
152
|
+
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
|
|
153
|
+
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
|
|
154
|
+
Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
|
|
155
|
+
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
|
|
156
|
+
Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
|
|
157
|
+
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
|
|
154
158
|
Provides-Extra: strategies
|
|
155
159
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "strategies"
|
|
156
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
160
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
|
|
157
161
|
Provides-Extra: test
|
|
158
162
|
Requires-Dist: click==8.1.8; extra == "test"
|
|
159
163
|
Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
|
|
160
|
-
Requires-Dist: coverage==7.
|
|
164
|
+
Requires-Dist: coverage==7.10.3; extra == "test"
|
|
161
165
|
Requires-Dist: fastapi; extra == "test"
|
|
162
166
|
Requires-Dist: numpy<2.0,>=1.17.2; extra == "test"
|
|
163
167
|
Requires-Dist: onnx<2.0,>=1.12.0; extra == "test"
|
|
164
168
|
Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "test"
|
|
169
|
+
Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "test"
|
|
165
170
|
Requires-Dist: pandas<3.0,>2.0; extra == "test"
|
|
166
171
|
Requires-Dist: psutil<8.0; extra == "test"
|
|
167
172
|
Requires-Dist: pytest-cov==6.2.1; extra == "test"
|
|
168
|
-
Requires-Dist: pytest-random-order==1.
|
|
173
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "test"
|
|
169
174
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "test"
|
|
170
175
|
Requires-Dist: pytest-timeout==2.4.0; extra == "test"
|
|
171
|
-
Requires-Dist: pytest==8.4.
|
|
176
|
+
Requires-Dist: pytest==8.4.1; extra == "test"
|
|
172
177
|
Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "test"
|
|
173
178
|
Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "test"
|
|
174
179
|
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "test"
|
|
175
180
|
Requires-Dist: uvicorn; extra == "test"
|
|
176
|
-
Provides-Extra: extra
|
|
177
|
-
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
|
|
178
|
-
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
|
|
179
|
-
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
|
|
180
|
-
Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
|
|
181
|
-
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
|
|
182
|
-
Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
|
|
183
|
-
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
|
|
184
181
|
Provides-Extra: examples
|
|
185
182
|
Requires-Dist: ipython[all]<9.0; extra == "examples"
|
|
186
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "examples"
|
|
187
183
|
Requires-Dist: requests<3.0; extra == "examples"
|
|
188
184
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "examples"
|
|
189
185
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "examples"
|
|
@@ -191,11 +187,10 @@ Provides-Extra: data
|
|
|
191
187
|
Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
|
|
192
188
|
Provides-Extra: all
|
|
193
189
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "all"
|
|
194
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
190
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
|
|
195
191
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "all"
|
|
196
192
|
Requires-Dist: ipython[all]<9.0; extra == "all"
|
|
197
193
|
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "all"
|
|
198
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "all"
|
|
199
194
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "all"
|
|
200
195
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "all"
|
|
201
196
|
Requires-Dist: requests<3.0; extra == "all"
|
|
@@ -207,25 +202,25 @@ Provides-Extra: dev
|
|
|
207
202
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "dev"
|
|
208
203
|
Requires-Dist: click==8.1.8; extra == "dev"
|
|
209
204
|
Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
|
|
210
|
-
Requires-Dist: coverage==7.
|
|
211
|
-
Requires-Dist: deepspeed<=0.9.3,>=0.
|
|
205
|
+
Requires-Dist: coverage==7.10.3; extra == "dev"
|
|
206
|
+
Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
|
|
212
207
|
Requires-Dist: fastapi; extra == "dev"
|
|
213
208
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "dev"
|
|
214
209
|
Requires-Dist: ipython[all]<9.0; extra == "dev"
|
|
215
210
|
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "dev"
|
|
216
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "dev"
|
|
217
211
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
|
|
218
212
|
Requires-Dist: numpy<2.0,>=1.17.2; extra == "dev"
|
|
219
213
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "dev"
|
|
220
214
|
Requires-Dist: onnx<2.0,>=1.12.0; extra == "dev"
|
|
221
215
|
Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "dev"
|
|
216
|
+
Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "dev"
|
|
222
217
|
Requires-Dist: pandas<3.0,>2.0; extra == "dev"
|
|
223
218
|
Requires-Dist: psutil<8.0; extra == "dev"
|
|
224
219
|
Requires-Dist: pytest-cov==6.2.1; extra == "dev"
|
|
225
|
-
Requires-Dist: pytest-random-order==1.
|
|
220
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
|
|
226
221
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "dev"
|
|
227
222
|
Requires-Dist: pytest-timeout==2.4.0; extra == "dev"
|
|
228
|
-
Requires-Dist: pytest==8.4.
|
|
223
|
+
Requires-Dist: pytest==8.4.1; extra == "dev"
|
|
229
224
|
Requires-Dist: requests<3.0; extra == "dev"
|
|
230
225
|
Requires-Dist: rich<15.0,>=12.3.0; extra == "dev"
|
|
231
226
|
Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "dev"
|
|
@@ -270,7 +265,7 @@ ______________________________________________________________________
|
|
|
270
265
|
<a href="#lightning-fabric-expert-control">Fabric</a> •
|
|
271
266
|
<a href="https://lightning.ai/">Lightning AI</a> •
|
|
272
267
|
<a href="#community">Community</a> •
|
|
273
|
-
<a href="https://pytorch-lightning.readthedocs.io/en/2.5.
|
|
268
|
+
<a href="https://pytorch-lightning.readthedocs.io/en/2.5.3">Docs</a>
|
|
274
269
|
</p>
|
|
275
270
|
|
|
276
271
|
<!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
|
|
@@ -307,6 +302,12 @@ ______________________________________________________________________
|
|
|
307
302
|
|
|
308
303
|
|
|
309
304
|
|
|
305
|
+
# Why PyTorch Lightning?
|
|
306
|
+
|
|
307
|
+
Training models in plain PyTorch is tedious and error-prone - you have to manually handle things like backprop, mixed precision, multi-GPU, and distributed training, often rewriting code for every new project. PyTorch Lightning organizes PyTorch code to automate those complexities so you can focus on your model and data, while keeping full control and scaling from CPU to multi-node without changing your core code. But if you want control of those things, you can still opt into more DIY.
|
|
308
|
+
|
|
309
|
+
Fun analogy: If PyTorch is Javascript, PyTorch Lightning is ReactJS or NextJS.
|
|
310
|
+
|
|
310
311
|
# Lightning has 2 core packages
|
|
311
312
|
|
|
312
313
|
[PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
|
|
@@ -795,7 +796,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
|
|
|
795
796
|
|
|
796
797
|
| System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
|
|
797
798
|
| :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
|
|
798
|
-
| Linux py3.9 \[GPUs\] | | | [](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
|
|
799
800
|
| Linux (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
800
801
|
| OSX (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
801
802
|
| Windows (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
@@ -55,6 +55,12 @@ ______________________________________________________________________
|
|
|
55
55
|
|
|
56
56
|
|
|
57
57
|
|
|
58
|
+
# Why PyTorch Lightning?
|
|
59
|
+
|
|
60
|
+
Training models in plain PyTorch is tedious and error-prone - you have to manually handle things like backprop, mixed precision, multi-GPU, and distributed training, often rewriting code for every new project. PyTorch Lightning organizes PyTorch code to automate those complexities so you can focus on your model and data, while keeping full control and scaling from CPU to multi-node without changing your core code. But if you want control of those things, you can still opt into more DIY.
|
|
61
|
+
|
|
62
|
+
Fun analogy: If PyTorch is Javascript, PyTorch Lightning is ReactJS or NextJS.
|
|
63
|
+
|
|
58
64
|
# Lightning has 2 core packages
|
|
59
65
|
|
|
60
66
|
[PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
PyYAML<7.0
|
|
1
|
+
PyYAML<7.0,>5.4
|
|
2
2
|
fsspec[http]<2026.0,>=2022.5.0
|
|
3
3
|
lightning-utilities<1.0,>=0.10.0
|
|
4
4
|
packaging<26.0,>=20.0
|
|
5
5
|
torch<3.0,>=2.1.0
|
|
6
|
-
torchmetrics<2.0
|
|
6
|
+
torchmetrics<2.0,>0.7.0
|
|
7
7
|
tqdm<5.0,>=4.57.0
|
|
8
|
-
typing-extensions<5.0
|
|
8
|
+
typing-extensions<5.0,>4.5.0
|
|
9
9
|
|
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
|
|
2
2
|
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
|
|
3
3
|
|
|
4
|
-
torch >=2.1.0, <2.
|
|
5
|
-
fsspec[http] >=2022.5.0, <2025.
|
|
4
|
+
torch >=2.1.0, <2.9.0
|
|
5
|
+
fsspec[http] >=2022.5.0, <2025.8.0
|
|
6
6
|
packaging >=20.0, <=25.0
|
|
7
|
-
typing-extensions
|
|
8
|
-
lightning-utilities >=0.10.0, <0.
|
|
7
|
+
typing-extensions >4.5.0, <4.15.0
|
|
8
|
+
lightning-utilities >=0.10.0, <0.16.0
|
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
|
|
2
2
|
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
|
|
3
3
|
|
|
4
|
-
torchvision >=0.16.0, <0.
|
|
5
|
-
torchmetrics >=0.10.0, <1.
|
|
6
|
-
lightning-utilities >=0.8.0, <0.15.0
|
|
4
|
+
torchvision >=0.16.0, <0.24.0
|
|
5
|
+
torchmetrics >=0.10.0, <1.9.0
|
|
@@ -5,5 +5,5 @@
|
|
|
5
5
|
|
|
6
6
|
# note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
|
|
7
7
|
# shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
|
|
8
|
-
deepspeed >=0.
|
|
8
|
+
deepspeed >=0.9.3, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
|
|
9
9
|
bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
coverage ==7.
|
|
1
|
+
coverage ==7.10.3
|
|
2
2
|
numpy >=1.17.2, <1.27.0
|
|
3
|
-
pytest ==8.4.
|
|
3
|
+
pytest ==8.4.1
|
|
4
4
|
pytest-cov ==6.2.1
|
|
5
5
|
pytest-timeout ==2.4.0
|
|
6
6
|
pytest-rerunfailures ==15.1
|
|
7
|
-
pytest-random-order ==1.
|
|
7
|
+
pytest-random-order ==1.2.0
|
|
8
8
|
click ==8.1.8
|
|
9
9
|
tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute
|
|
@@ -1,11 +1,11 @@
|
|
|
1
1
|
# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
|
|
2
2
|
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
|
|
3
3
|
|
|
4
|
-
torch >=2.1.0, <2.
|
|
4
|
+
torch >=2.1.0, <2.9.0
|
|
5
5
|
tqdm >=4.57.0, <4.68.0
|
|
6
|
-
PyYAML
|
|
7
|
-
fsspec[http] >=2022.5.0, <2025.
|
|
8
|
-
torchmetrics
|
|
6
|
+
PyYAML >5.4, <6.1.0
|
|
7
|
+
fsspec[http] >=2022.5.0, <2025.8.0
|
|
8
|
+
torchmetrics >0.7.0, <1.9.0
|
|
9
9
|
packaging >=20.0, <=25.0
|
|
10
|
-
typing-extensions
|
|
11
|
-
lightning-utilities >=0.10.0, <0.
|
|
10
|
+
typing-extensions >4.5.0, <4.15.0
|
|
11
|
+
lightning-utilities >=0.10.0, <0.16.0
|
|
@@ -4,4 +4,6 @@ nbformat # used for generate empty notebook
|
|
|
4
4
|
ipython[notebook] <8.19.0
|
|
5
5
|
setuptools<81.0 # workaround for `error in ipython setup command: use_2to3 is invalid.`
|
|
6
6
|
|
|
7
|
-
|
|
7
|
+
onnxscript >= 0.2.2, <0.4.0
|
|
8
|
+
|
|
9
|
+
#-r ../../_notebooks/.actions/requires.txt
|
|
@@ -2,7 +2,6 @@
|
|
|
2
2
|
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
|
|
3
3
|
|
|
4
4
|
requests <2.33.0
|
|
5
|
-
torchvision >=0.16.0, <0.
|
|
5
|
+
torchvision >=0.16.0, <0.24.0
|
|
6
6
|
ipython[all] <8.19.0
|
|
7
|
-
torchmetrics >=0.10.0, <1.
|
|
8
|
-
lightning-utilities >=0.8.0, <0.15.0
|
|
7
|
+
torchmetrics >=0.10.0, <1.9.0
|
|
@@ -6,6 +6,6 @@ matplotlib>3.1, <3.10.0
|
|
|
6
6
|
omegaconf >=2.2.3, <2.4.0
|
|
7
7
|
hydra-core >=1.2.0, <1.4.0
|
|
8
8
|
jsonargparse[signatures,jsonnet] >=4.39.0, <4.41.0
|
|
9
|
-
rich >=12.3.0, <14.
|
|
9
|
+
rich >=12.3.0, <14.2.0
|
|
10
10
|
tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute
|
|
11
11
|
bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
|
|
@@ -3,4 +3,4 @@
|
|
|
3
3
|
|
|
4
4
|
# note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
|
|
5
5
|
# shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
|
|
6
|
-
deepspeed >=0.
|
|
6
|
+
deepspeed >=0.9.3, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
coverage ==7.
|
|
2
|
-
pytest ==8.4.
|
|
1
|
+
coverage ==7.10.3
|
|
2
|
+
pytest ==8.4.1
|
|
3
3
|
pytest-cov ==6.2.1
|
|
4
4
|
pytest-timeout ==2.4.0
|
|
5
5
|
pytest-rerunfailures ==15.1
|
|
6
|
-
pytest-random-order ==1.
|
|
6
|
+
pytest-random-order ==1.2.0
|
|
7
7
|
|
|
8
8
|
# needed in tests
|
|
9
9
|
cloudpickle >=1.3, <3.2.0
|
|
@@ -11,9 +11,10 @@ scikit-learn >0.22.1, <1.7.0
|
|
|
11
11
|
numpy >=1.17.2, <1.27.0
|
|
12
12
|
onnx >=1.12.0, <1.19.0
|
|
13
13
|
onnxruntime >=1.12.0, <1.21.0
|
|
14
|
+
onnxscript >= 0.2.2, <0.4.0
|
|
14
15
|
psutil <7.0.1 # for `DeviceStatsMonitor`
|
|
15
16
|
pandas >2.0, <2.4.0 # needed in benchmarks
|
|
16
17
|
fastapi # for `ServableModuleValidator` # not setting version as re-defined in App
|
|
17
18
|
uvicorn # for `ServableModuleValidator` # not setting version as re-defined in App
|
|
18
19
|
|
|
19
|
-
tensorboard >=2.9.1, <2.
|
|
20
|
+
tensorboard >=2.9.1, <2.21.0 # for `TensorBoardLogger`
|
|
@@ -15,7 +15,7 @@
|
|
|
15
15
|
import time
|
|
16
16
|
|
|
17
17
|
__author__ = "Lightning AI et al."
|
|
18
|
-
__author_email__ = "
|
|
18
|
+
__author_email__ = "developer@lightning.ai"
|
|
19
19
|
__license__ = "Apache-2.0"
|
|
20
20
|
__copyright__ = f"Copyright (c) 2018-{time.strftime('%Y')}, {__author__}."
|
|
21
21
|
__homepage__ = "https://github.com/Lightning-AI/lightning"
|
|
@@ -6,6 +6,19 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
|
|
|
6
6
|
|
|
7
7
|
---
|
|
8
8
|
|
|
9
|
+
## [2.5.3] - 2025-08-DD
|
|
10
|
+
|
|
11
|
+
### Changed
|
|
12
|
+
|
|
13
|
+
- Enable "auto" for `devices` and `accelerator` as CLI arguments ([#20913](https://github.com/Lightning-AI/pytorch-lightning/pull/20913))
|
|
14
|
+
- Raise ValueError when seed is `out-of-bounds` or `cannot be cast to int` ([#21029](https://github.com/Lightning-AI/pytorch-lightning/pull/21029))
|
|
15
|
+
|
|
16
|
+
### Fixed
|
|
17
|
+
|
|
18
|
+
- Fixed XLA strategy to add support for `global_ordinal`, `local_ordinal`, `world_size` which came instead of deprecated methods ([#20852](https://github.com/Lightning-AI/pytorch-lightning/issues/20852))
|
|
19
|
+
- Fixed remove extra `name` parameter in accelerator registry decorator ([#20975](https://github.com/Lightning-AI/pytorch-lightning/pull/20975))
|
|
20
|
+
|
|
21
|
+
|
|
9
22
|
## [2.5.2] - 2025-3-20
|
|
10
23
|
|
|
11
24
|
### Changed
|
|
@@ -73,14 +73,14 @@ class _AcceleratorRegistry(dict):
|
|
|
73
73
|
data["description"] = description
|
|
74
74
|
data["init_params"] = init_params
|
|
75
75
|
|
|
76
|
-
def do_register(
|
|
76
|
+
def do_register(accelerator: Callable) -> Callable:
|
|
77
77
|
data["accelerator"] = accelerator
|
|
78
78
|
data["accelerator_name"] = name
|
|
79
79
|
self[name] = data
|
|
80
80
|
return accelerator
|
|
81
81
|
|
|
82
82
|
if accelerator is not None:
|
|
83
|
-
return do_register(
|
|
83
|
+
return do_register(accelerator)
|
|
84
84
|
|
|
85
85
|
return do_register
|
|
86
86
|
|
|
@@ -25,7 +25,7 @@ from lightning.fabric.accelerators import CPUAccelerator, CUDAAccelerator, MPSAc
|
|
|
25
25
|
from lightning.fabric.plugins.precision.precision import _PRECISION_INPUT_STR, _PRECISION_INPUT_STR_ALIAS
|
|
26
26
|
from lightning.fabric.strategies import STRATEGY_REGISTRY
|
|
27
27
|
from lightning.fabric.utilities.consolidate_checkpoint import _process_cli_args
|
|
28
|
-
from lightning.fabric.utilities.device_parser import _parse_gpu_ids
|
|
28
|
+
from lightning.fabric.utilities.device_parser import _parse_gpu_ids, _select_auto_accelerator
|
|
29
29
|
from lightning.fabric.utilities.distributed import _suggested_max_num_threads
|
|
30
30
|
from lightning.fabric.utilities.load import _load_distributed_checkpoint
|
|
31
31
|
|
|
@@ -34,7 +34,7 @@ _log = logging.getLogger(__name__)
|
|
|
34
34
|
_CLICK_AVAILABLE = RequirementCache("click")
|
|
35
35
|
_LIGHTNING_SDK_AVAILABLE = RequirementCache("lightning_sdk")
|
|
36
36
|
|
|
37
|
-
_SUPPORTED_ACCELERATORS = ("cpu", "gpu", "cuda", "mps", "tpu")
|
|
37
|
+
_SUPPORTED_ACCELERATORS = ("cpu", "gpu", "cuda", "mps", "tpu", "auto")
|
|
38
38
|
|
|
39
39
|
|
|
40
40
|
def _get_supported_strategies() -> list[str]:
|
|
@@ -187,6 +187,14 @@ def _set_env_variables(args: Namespace) -> None:
|
|
|
187
187
|
|
|
188
188
|
def _get_num_processes(accelerator: str, devices: str) -> int:
|
|
189
189
|
"""Parse the `devices` argument to determine how many processes need to be launched on the current machine."""
|
|
190
|
+
|
|
191
|
+
if accelerator == "auto" or accelerator is None:
|
|
192
|
+
accelerator = _select_auto_accelerator()
|
|
193
|
+
if devices == "auto":
|
|
194
|
+
if accelerator == "cuda" or accelerator == "mps" or accelerator == "cpu":
|
|
195
|
+
devices = "1"
|
|
196
|
+
else:
|
|
197
|
+
raise ValueError(f"Cannot default to '1' device for accelerator='{accelerator}'")
|
|
190
198
|
if accelerator == "gpu":
|
|
191
199
|
parsed_devices = _parse_gpu_ids(devices, include_cuda=True, include_mps=True)
|
|
192
200
|
elif accelerator == "cuda":
|
|
@@ -66,6 +66,11 @@ class XLAEnvironment(ClusterEnvironment):
|
|
|
66
66
|
The output is cached for performance.
|
|
67
67
|
|
|
68
68
|
"""
|
|
69
|
+
if _XLA_GREATER_EQUAL_2_1:
|
|
70
|
+
from torch_xla import runtime as xr
|
|
71
|
+
|
|
72
|
+
return xr.world_size()
|
|
73
|
+
|
|
69
74
|
import torch_xla.core.xla_model as xm
|
|
70
75
|
|
|
71
76
|
return xm.xrt_world_size()
|
|
@@ -82,6 +87,11 @@ class XLAEnvironment(ClusterEnvironment):
|
|
|
82
87
|
The output is cached for performance.
|
|
83
88
|
|
|
84
89
|
"""
|
|
90
|
+
if _XLA_GREATER_EQUAL_2_1:
|
|
91
|
+
from torch_xla import runtime as xr
|
|
92
|
+
|
|
93
|
+
return xr.global_ordinal()
|
|
94
|
+
|
|
85
95
|
import torch_xla.core.xla_model as xm
|
|
86
96
|
|
|
87
97
|
return xm.get_ordinal()
|
|
@@ -98,6 +108,11 @@ class XLAEnvironment(ClusterEnvironment):
|
|
|
98
108
|
The output is cached for performance.
|
|
99
109
|
|
|
100
110
|
"""
|
|
111
|
+
if _XLA_GREATER_EQUAL_2_1:
|
|
112
|
+
from torch_xla import runtime as xr
|
|
113
|
+
|
|
114
|
+
return xr.local_ordinal()
|
|
115
|
+
|
|
101
116
|
import torch_xla.core.xla_model as xm
|
|
102
117
|
|
|
103
118
|
return xm.get_local_ordinal()
|
|
@@ -256,10 +256,12 @@ def _import_bitsandbytes() -> ModuleType:
|
|
|
256
256
|
if int8params.has_fp16_weights:
|
|
257
257
|
int8params.data = B
|
|
258
258
|
else:
|
|
259
|
-
|
|
259
|
+
# bitsandbytes >= 0.45 supports an improved API
|
|
260
|
+
if hasattr(bnb.functional, "int8_vectorwise_quant"):
|
|
261
|
+
CB, SCB, _ = bnb.functional.int8_vectorwise_quant(B)
|
|
262
|
+
else: # old method is deprecated in 0.45, removed in 0.46+.
|
|
260
263
|
CB, _, SCB, _, _ = bnb.functional.double_quant(B)
|
|
261
|
-
|
|
262
|
-
CB, SCB = bnb.functional.int8_double_quant(B)
|
|
264
|
+
|
|
263
265
|
int8params.data = CB
|
|
264
266
|
setattr(int8params, "CB", CB)
|
|
265
267
|
setattr(int8params, "SCB", SCB)
|