lightning 2.5.2__tar.gz → 2.5.4__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 → lightning-2.5.4}/.actions/assistant.py +0 -41
- {lightning-2.5.2/src/lightning.egg-info → lightning-2.5.4}/PKG-INFO +88 -80
- {lightning-2.5.2 → lightning-2.5.4}/README.md +6 -0
- {lightning-2.5.2 → lightning-2.5.4}/pyproject.toml +2 -0
- {lightning-2.5.2 → lightning-2.5.4}/requirements/base.txt +3 -3
- {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/base.txt +4 -4
- {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/examples.txt +2 -3
- {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/strategies.txt +1 -1
- lightning-2.5.4/requirements/fabric/test.txt +10 -0
- {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/base.txt +6 -6
- {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/docs.txt +4 -2
- {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/examples.txt +2 -3
- {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/extra.txt +2 -2
- {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/strategies.txt +1 -1
- {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/test.txt +9 -8
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__about__.py +1 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/CHANGELOG.md +20 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/registry.py +2 -2
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/cli.py +10 -2
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/torch_collective.py +11 -14
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/xla.py +15 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/bitsandbytes.py +5 -3
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/deepspeed.py +1 -5
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/consolidate_checkpoint.py +6 -5
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/device_parser.py +15 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/imports.py +2 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/seed.py +4 -5
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/spike.py +1 -4
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/testing/_runif.py +7 -5
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/throughput.py +23 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/CHANGELOG.md +35 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/lr_finder.py +1 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/model_checkpoint.py +49 -6
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/rich_progress.py +40 -7
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/rich_model_summary.py +5 -2
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/module.py +62 -11
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/optimizer.py +1 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/transformer.py +13 -4
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/loop.py +8 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/manual.py +2 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/training_epoch_loop.py +5 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/async_plugin.py +40 -5
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/advanced.py +6 -4
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/call.py +2 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +6 -11
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/data_connector.py +8 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +3 -2
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +1 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/trainer.py +3 -4
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/lr_finder.py +26 -33
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/imports.py +4 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_summary/model_summary.py +17 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +1 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/parsing.py +2 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/testing/_runif.py +11 -4
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +3 -2
- lightning-2.5.4/src/lightning/version.info +1 -0
- {lightning-2.5.2 → lightning-2.5.4/src/lightning.egg-info}/PKG-INFO +88 -80
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/SOURCES.txt +1 -1
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/requires.txt +78 -60
- lightning-2.5.4/src/version.info +1 -0
- lightning-2.5.2/requirements/fabric/test.txt +0 -9
- 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.4}/CITATION.cff +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/LICENSE +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/docs.txt +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/setup.cfg +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/setup.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__setup__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__version__.py +0 -0
- /lightning-2.5.2/src/lightning/ffd6261c9058703d8c54093b4f93e2b4008030713613e504 → /lightning-2.5.4/src/lightning/d4cfc24c31e95ba11b2250aa0b5d0d903d59b77c4fbee1d2 +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/data/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/_graveyard/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/_graveyard/tpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/accelerator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/cpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/cuda.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/mps.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/fabric.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/csv_logs.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/logger.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/tensorboard.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/single_device.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/lightning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/lsf.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/torch_io.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/amp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/double.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/half.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/precision.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/transformer_engine.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/utils.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/ddp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/dp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/multiprocessing.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/subprocess_script.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/model_parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/registry.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/single_device.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/single_xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/strategy.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/xla_fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/apply_func.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/cloud_io.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/data.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/device_dtype_mixin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/distributed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/enums.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/exceptions.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/init.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/load.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/logger.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/optimizer.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/rank_zero.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/registry.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/types.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/warnings.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/wrappers.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/py.typed +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/precision.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/cpu.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/cuda.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/mps.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/callback.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/device_stats_monitor.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/early_stopping.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/finetuning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/lr_monitor.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/model_summary.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/prediction_writer.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/progress_bar.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/pruning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/spike.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/throughput_monitor.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/timer.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/cli.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/datamodule.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/hooks.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/mixins/hparams_mixin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/saving.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/boring_classes.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/lstm.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/comet.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/csv_logs.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/logger.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/mlflow.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/neptune.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/tensorboard.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/utilities.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/wandb.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/evaluation_loop.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/fetchers.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/fit_loop.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/automatic.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/prediction_loop.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/progress.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/utilities.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/overrides/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/overrides/distributed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/wrapper.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/amp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/double.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/half.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/precision.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/base.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/profiler.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/pytorch.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/simple.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/serve/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/serve/servable_module.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/serve/servable_module_validator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/ddp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/fsdp.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/model_parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/parallel.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/single_device.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/single_xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/strategy.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/xla.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/callback_connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/signal_connector.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/setup.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/states.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/batch_size_scaling.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/tuning.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/_pytree.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/argparse.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/combined_loader.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/compile.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/data.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/deepspeed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/enums.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/exceptions.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/grads.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/memory.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/migration/utils.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_helpers.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_registry.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/seed.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/types.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/warnings.py +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/dependency_links.txt +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/entry_points.txt +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/not-zip-safe +0 -0
- {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/top_level.txt +0 -0
|
@@ -341,47 +341,6 @@ def create_mirror_package(source_dir: str, package_mapping: dict[str, str]) -> N
|
|
|
341
341
|
|
|
342
342
|
|
|
343
343
|
class AssistantCLI:
|
|
344
|
-
@staticmethod
|
|
345
|
-
def requirements_prune_pkgs(packages: Sequence[str], req_files: Sequence[str] = REQUIREMENT_FILES_ALL) -> None:
|
|
346
|
-
"""Remove some packages from given requirement files."""
|
|
347
|
-
if isinstance(req_files, str):
|
|
348
|
-
req_files = [req_files]
|
|
349
|
-
for req in req_files:
|
|
350
|
-
AssistantCLI._prune_packages(req, packages)
|
|
351
|
-
|
|
352
|
-
@staticmethod
|
|
353
|
-
def _prune_packages(req_file: str, packages: Sequence[str]) -> None:
|
|
354
|
-
"""Remove some packages from given requirement files."""
|
|
355
|
-
path = Path(req_file)
|
|
356
|
-
assert path.exists()
|
|
357
|
-
text = path.read_text()
|
|
358
|
-
lines = text.splitlines()
|
|
359
|
-
final = []
|
|
360
|
-
for line in lines:
|
|
361
|
-
ln_ = line.strip()
|
|
362
|
-
if not ln_ or ln_.startswith("#"):
|
|
363
|
-
final.append(line)
|
|
364
|
-
continue
|
|
365
|
-
req = list(_parse_requirements([ln_]))[0]
|
|
366
|
-
if req.name not in packages:
|
|
367
|
-
final.append(line)
|
|
368
|
-
print(final)
|
|
369
|
-
path.write_text("\n".join(final) + "\n")
|
|
370
|
-
|
|
371
|
-
@staticmethod
|
|
372
|
-
def _replace_min(fname: str) -> None:
|
|
373
|
-
with open(fname, encoding="utf-8") as fopen:
|
|
374
|
-
req = fopen.read().replace(">=", "==")
|
|
375
|
-
with open(fname, "w", encoding="utf-8") as fwrite:
|
|
376
|
-
fwrite.write(req)
|
|
377
|
-
|
|
378
|
-
@staticmethod
|
|
379
|
-
def replace_oldest_ver(requirement_fnames: Sequence[str] = REQUIREMENT_FILES_ALL) -> None:
|
|
380
|
-
"""Replace the min package version by fixed one."""
|
|
381
|
-
for fname in requirement_fnames:
|
|
382
|
-
print(fname)
|
|
383
|
-
AssistantCLI._replace_min(fname)
|
|
384
|
-
|
|
385
344
|
@staticmethod
|
|
386
345
|
def copy_replace_imports(
|
|
387
346
|
source_dir: str,
|
|
@@ -1,11 +1,11 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: lightning
|
|
3
|
-
Version: 2.5.
|
|
3
|
+
Version: 2.5.4
|
|
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,91 @@ 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.
|
|
48
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (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.5; extra == "pytorch-test"
|
|
44
52
|
Requires-Dist: fastapi; extra == "pytorch-test"
|
|
45
|
-
Requires-Dist: numpy<2.0
|
|
46
|
-
Requires-Dist: onnx<2.0
|
|
53
|
+
Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-test"
|
|
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.1.0; 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
|
-
Requires-Dist: tensorboard<3.0,>=2.
|
|
65
|
+
Requires-Dist: tensorboard<3.0,>=2.11; 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.
|
|
74
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
|
|
75
75
|
Provides-Extra: fabric-test
|
|
76
|
-
Requires-Dist: click==8.1.8; extra == "fabric-test"
|
|
77
|
-
Requires-Dist:
|
|
78
|
-
Requires-Dist:
|
|
76
|
+
Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-test"
|
|
77
|
+
Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-test"
|
|
78
|
+
Requires-Dist: coverage==7.10.5; extra == "fabric-test"
|
|
79
|
+
Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-test"
|
|
79
80
|
Requires-Dist: pytest-cov==6.2.1; extra == "fabric-test"
|
|
80
|
-
Requires-Dist: pytest-random-order==1.
|
|
81
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-test"
|
|
81
82
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-test"
|
|
82
83
|
Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-test"
|
|
83
|
-
Requires-Dist: pytest==8.4.
|
|
84
|
-
Requires-Dist: tensorboardX<3.0,>=2.
|
|
84
|
+
Requires-Dist: pytest==8.4.1; extra == "fabric-test"
|
|
85
|
+
Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-test"
|
|
85
86
|
Provides-Extra: fabric-examples
|
|
86
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-examples"
|
|
87
87
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-examples"
|
|
88
88
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-examples"
|
|
89
89
|
Provides-Extra: fabric-all
|
|
90
90
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-all"
|
|
91
|
-
Requires-Dist: deepspeed<=0.
|
|
92
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-all"
|
|
91
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
|
|
93
92
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-all"
|
|
94
93
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-all"
|
|
95
94
|
Provides-Extra: fabric-dev
|
|
96
95
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-dev"
|
|
97
|
-
Requires-Dist: click==8.1.8; extra == "fabric-dev"
|
|
98
|
-
Requires-Dist:
|
|
99
|
-
Requires-Dist:
|
|
100
|
-
Requires-Dist:
|
|
101
|
-
Requires-Dist: numpy<2.0,>=1.
|
|
96
|
+
Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-dev"
|
|
97
|
+
Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-dev"
|
|
98
|
+
Requires-Dist: coverage==7.10.5; extra == "fabric-dev"
|
|
99
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
|
|
100
|
+
Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-dev"
|
|
102
101
|
Requires-Dist: pytest-cov==6.2.1; extra == "fabric-dev"
|
|
103
|
-
Requires-Dist: pytest-random-order==1.
|
|
102
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-dev"
|
|
104
103
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-dev"
|
|
105
104
|
Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-dev"
|
|
106
|
-
Requires-Dist: pytest==8.4.
|
|
107
|
-
Requires-Dist: tensorboardX<3.0,>=2.
|
|
105
|
+
Requires-Dist: pytest==8.4.1; extra == "fabric-dev"
|
|
106
|
+
Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-dev"
|
|
108
107
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-dev"
|
|
109
108
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-dev"
|
|
110
109
|
Provides-Extra: pytorch-all
|
|
111
110
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-all"
|
|
112
|
-
Requires-Dist: deepspeed<=0.
|
|
111
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
|
|
113
112
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-all"
|
|
114
113
|
Requires-Dist: ipython[all]<9.0; extra == "pytorch-all"
|
|
115
114
|
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
115
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-all"
|
|
118
116
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-all"
|
|
119
117
|
Requires-Dist: requests<3.0; extra == "pytorch-all"
|
|
@@ -124,66 +122,68 @@ Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-all"
|
|
|
124
122
|
Provides-Extra: pytorch-dev
|
|
125
123
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-dev"
|
|
126
124
|
Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-dev"
|
|
127
|
-
Requires-Dist: coverage==7.
|
|
128
|
-
Requires-Dist: deepspeed<=0.
|
|
125
|
+
Requires-Dist: coverage==7.10.5; extra == "pytorch-dev"
|
|
126
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
|
|
129
127
|
Requires-Dist: fastapi; extra == "pytorch-dev"
|
|
130
128
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-dev"
|
|
131
129
|
Requires-Dist: ipython[all]<9.0; extra == "pytorch-dev"
|
|
132
130
|
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
131
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
|
|
135
|
-
Requires-Dist: numpy<2.0
|
|
132
|
+
Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-dev"
|
|
136
133
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-dev"
|
|
137
|
-
Requires-Dist: onnx<2.0
|
|
134
|
+
Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-dev"
|
|
138
135
|
Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-dev"
|
|
136
|
+
Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-dev"
|
|
139
137
|
Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-dev"
|
|
140
138
|
Requires-Dist: psutil<8.0; extra == "pytorch-dev"
|
|
141
139
|
Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-dev"
|
|
142
|
-
Requires-Dist: pytest-random-order==1.
|
|
140
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-dev"
|
|
143
141
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-dev"
|
|
144
142
|
Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-dev"
|
|
145
|
-
Requires-Dist: pytest==8.4.
|
|
143
|
+
Requires-Dist: pytest==8.4.1; extra == "pytorch-dev"
|
|
146
144
|
Requires-Dist: requests<3.0; extra == "pytorch-dev"
|
|
147
145
|
Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-dev"
|
|
148
146
|
Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-dev"
|
|
149
|
-
Requires-Dist: tensorboard<3.0,>=2.
|
|
147
|
+
Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-dev"
|
|
150
148
|
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-dev"
|
|
151
149
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-dev"
|
|
152
150
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-dev"
|
|
153
151
|
Requires-Dist: uvicorn; extra == "pytorch-dev"
|
|
152
|
+
Provides-Extra: extra
|
|
153
|
+
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
|
|
154
|
+
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
|
|
155
|
+
Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
|
|
156
|
+
Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
|
|
157
|
+
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
|
|
158
|
+
Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
|
|
159
|
+
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
|
|
154
160
|
Provides-Extra: strategies
|
|
155
161
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "strategies"
|
|
156
|
-
Requires-Dist: deepspeed<=0.
|
|
162
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
|
|
157
163
|
Provides-Extra: test
|
|
158
|
-
Requires-Dist: click==8.1.8; extra == "test"
|
|
164
|
+
Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "test"
|
|
165
|
+
Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "test"
|
|
159
166
|
Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
|
|
160
|
-
Requires-Dist: coverage==7.
|
|
167
|
+
Requires-Dist: coverage==7.10.5; extra == "test"
|
|
161
168
|
Requires-Dist: fastapi; extra == "test"
|
|
162
|
-
Requires-Dist: numpy<2.0
|
|
163
|
-
Requires-Dist:
|
|
169
|
+
Requires-Dist: numpy<2.0,>1.20.0; extra == "test"
|
|
170
|
+
Requires-Dist: numpy<2.0,>=1.21.0; extra == "test"
|
|
171
|
+
Requires-Dist: onnx<2.0,>1.12.0; extra == "test"
|
|
164
172
|
Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "test"
|
|
173
|
+
Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "test"
|
|
165
174
|
Requires-Dist: pandas<3.0,>2.0; extra == "test"
|
|
166
175
|
Requires-Dist: psutil<8.0; extra == "test"
|
|
167
176
|
Requires-Dist: pytest-cov==6.2.1; extra == "test"
|
|
168
|
-
Requires-Dist: pytest-random-order==1.
|
|
177
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "test"
|
|
169
178
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "test"
|
|
170
179
|
Requires-Dist: pytest-timeout==2.4.0; extra == "test"
|
|
171
|
-
Requires-Dist: pytest==8.4.
|
|
180
|
+
Requires-Dist: pytest==8.4.1; extra == "test"
|
|
172
181
|
Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "test"
|
|
173
|
-
Requires-Dist: tensorboard<3.0,>=2.
|
|
174
|
-
Requires-Dist: tensorboardX<3.0,>=2.
|
|
182
|
+
Requires-Dist: tensorboard<3.0,>=2.11; extra == "test"
|
|
183
|
+
Requires-Dist: tensorboardX<3.0,>=2.6; extra == "test"
|
|
175
184
|
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
185
|
Provides-Extra: examples
|
|
185
186
|
Requires-Dist: ipython[all]<9.0; extra == "examples"
|
|
186
|
-
Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "examples"
|
|
187
187
|
Requires-Dist: requests<3.0; extra == "examples"
|
|
188
188
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "examples"
|
|
189
189
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "examples"
|
|
@@ -191,11 +191,10 @@ Provides-Extra: data
|
|
|
191
191
|
Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
|
|
192
192
|
Provides-Extra: all
|
|
193
193
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "all"
|
|
194
|
-
Requires-Dist: deepspeed<=0.
|
|
194
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
|
|
195
195
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "all"
|
|
196
196
|
Requires-Dist: ipython[all]<9.0; extra == "all"
|
|
197
197
|
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
198
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "all"
|
|
200
199
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "all"
|
|
201
200
|
Requires-Dist: requests<3.0; extra == "all"
|
|
@@ -205,32 +204,35 @@ Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "all"
|
|
|
205
204
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "all"
|
|
206
205
|
Provides-Extra: dev
|
|
207
206
|
Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "dev"
|
|
208
|
-
Requires-Dist: click==8.1.8; extra == "dev"
|
|
207
|
+
Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "dev"
|
|
208
|
+
Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "dev"
|
|
209
209
|
Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
|
|
210
|
-
Requires-Dist: coverage==7.
|
|
211
|
-
Requires-Dist: deepspeed<=0.
|
|
210
|
+
Requires-Dist: coverage==7.10.5; extra == "dev"
|
|
211
|
+
Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
|
|
212
212
|
Requires-Dist: fastapi; extra == "dev"
|
|
213
213
|
Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "dev"
|
|
214
214
|
Requires-Dist: ipython[all]<9.0; extra == "dev"
|
|
215
215
|
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
216
|
Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
|
|
218
|
-
Requires-Dist: numpy<2.0
|
|
217
|
+
Requires-Dist: numpy<2.0,>1.20.0; extra == "dev"
|
|
218
|
+
Requires-Dist: numpy<2.0,>=1.21.0; extra == "dev"
|
|
219
219
|
Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "dev"
|
|
220
|
-
Requires-Dist: onnx<2.0
|
|
220
|
+
Requires-Dist: onnx<2.0,>1.12.0; extra == "dev"
|
|
221
221
|
Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "dev"
|
|
222
|
+
Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "dev"
|
|
222
223
|
Requires-Dist: pandas<3.0,>2.0; extra == "dev"
|
|
223
224
|
Requires-Dist: psutil<8.0; extra == "dev"
|
|
224
225
|
Requires-Dist: pytest-cov==6.2.1; extra == "dev"
|
|
225
|
-
Requires-Dist: pytest-random-order==1.
|
|
226
|
+
Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
|
|
226
227
|
Requires-Dist: pytest-rerunfailures==15.1; extra == "dev"
|
|
227
228
|
Requires-Dist: pytest-timeout==2.4.0; extra == "dev"
|
|
228
|
-
Requires-Dist: pytest==8.4.
|
|
229
|
+
Requires-Dist: pytest==8.4.1; extra == "dev"
|
|
229
230
|
Requires-Dist: requests<3.0; extra == "dev"
|
|
230
231
|
Requires-Dist: rich<15.0,>=12.3.0; extra == "dev"
|
|
231
232
|
Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "dev"
|
|
232
|
-
Requires-Dist: tensorboard<3.0,>=2.
|
|
233
|
+
Requires-Dist: tensorboard<3.0,>=2.11; extra == "dev"
|
|
233
234
|
Requires-Dist: tensorboardX<3.0,>=2.2; extra == "dev"
|
|
235
|
+
Requires-Dist: tensorboardX<3.0,>=2.6; extra == "dev"
|
|
234
236
|
Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "dev"
|
|
235
237
|
Requires-Dist: torchvision<1.0,>=0.16.0; extra == "dev"
|
|
236
238
|
Requires-Dist: uvicorn; extra == "dev"
|
|
@@ -270,7 +272,7 @@ ______________________________________________________________________
|
|
|
270
272
|
<a href="#lightning-fabric-expert-control">Fabric</a> •
|
|
271
273
|
<a href="https://lightning.ai/">Lightning AI</a> •
|
|
272
274
|
<a href="#community">Community</a> •
|
|
273
|
-
<a href="https://pytorch-lightning.readthedocs.io/en/2.5.
|
|
275
|
+
<a href="https://pytorch-lightning.readthedocs.io/en/2.5.4">Docs</a>
|
|
274
276
|
</p>
|
|
275
277
|
|
|
276
278
|
<!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
|
|
@@ -307,6 +309,12 @@ ______________________________________________________________________
|
|
|
307
309
|
|
|
308
310
|
|
|
309
311
|
|
|
312
|
+
# Why PyTorch Lightning?
|
|
313
|
+
|
|
314
|
+
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.
|
|
315
|
+
|
|
316
|
+
Fun analogy: If PyTorch is Javascript, PyTorch Lightning is ReactJS or NextJS.
|
|
317
|
+
|
|
310
318
|
# Lightning has 2 core packages
|
|
311
319
|
|
|
312
320
|
[PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
|
|
@@ -795,7 +803,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
|
|
|
795
803
|
|
|
796
804
|
| System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
|
|
797
805
|
| :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
|
|
798
|
-
| Linux py3.9 \[GPUs\] | | | [](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
|
|
799
807
|
| 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
808
|
| 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
809
|
| 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).
|
|
@@ -179,7 +179,9 @@ markers = [
|
|
|
179
179
|
"cloud: Run the cloud tests for example",
|
|
180
180
|
]
|
|
181
181
|
filterwarnings = [
|
|
182
|
+
# "error::DeprecationWarning",
|
|
182
183
|
"error::FutureWarning",
|
|
184
|
+
"ignore::FutureWarning:onnxscript", # Temporary ignore until onnxscript is updated
|
|
183
185
|
]
|
|
184
186
|
xfail_strict = true
|
|
185
187
|
junit_duration_report = "call"
|
|
@@ -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.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin" # strict
|
|
9
9
|
bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
coverage ==7.10.5
|
|
2
|
+
numpy >=1.21.0, <1.27.0
|
|
3
|
+
pytest ==8.4.1
|
|
4
|
+
pytest-cov ==6.2.1
|
|
5
|
+
pytest-timeout ==2.4.0
|
|
6
|
+
pytest-rerunfailures ==15.1
|
|
7
|
+
pytest-random-order ==1.2.0
|
|
8
|
+
click ==8.1.8; python_version < "3.11"
|
|
9
|
+
click ==8.2.1; python_version > "3.10"
|
|
10
|
+
tensorboardX >=2.6, <2.7.0 # todo: relax it back to `>=2.2` after fixing tests
|
|
@@ -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
|
|
@@ -1,7 +1,9 @@
|
|
|
1
1
|
-r ../docs.txt
|
|
2
2
|
|
|
3
3
|
nbformat # used for generate empty notebook
|
|
4
|
-
ipython[notebook] <
|
|
4
|
+
ipython[notebook] <9.5.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.5.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
|
|
@@ -2,10 +2,10 @@
|
|
|
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
|
# extended list of package dependencies to reach full functionality
|
|
5
|
-
matplotlib>3.1, <3.
|
|
5
|
+
matplotlib>3.1, <3.11.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.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin" # strict
|
|
@@ -1,19 +1,20 @@
|
|
|
1
|
-
coverage ==7.
|
|
2
|
-
pytest ==8.4.
|
|
1
|
+
coverage ==7.10.5
|
|
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
|
|
10
|
-
scikit-learn >0.22.1, <1.
|
|
11
|
-
numpy
|
|
12
|
-
onnx
|
|
13
|
-
onnxruntime >=1.12.0, <1.
|
|
10
|
+
scikit-learn >0.22.1, <1.8.0
|
|
11
|
+
numpy >1.20.0, <1.27.0
|
|
12
|
+
onnx >1.12.0, <1.19.0
|
|
13
|
+
onnxruntime >=1.12.0, <1.23.0
|
|
14
|
+
onnxscript >= 0.1.0, < 0.5.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.
|
|
20
|
+
tensorboard >=2.11, <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,26 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
|
|
|
6
6
|
|
|
7
7
|
---
|
|
8
8
|
|
|
9
|
+
## [2.5.4] - 2025-08-29
|
|
10
|
+
|
|
11
|
+
### Changed
|
|
12
|
+
|
|
13
|
+
- Added support for NVIDIA H200 GPUs in `get_available_flops` ([#20913](https://github.com/Lightning-AI/pytorch-lightning/pull/21119))
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
## [2.5.3] - 2025-08-13
|
|
17
|
+
|
|
18
|
+
### Changed
|
|
19
|
+
|
|
20
|
+
- Enable "auto" for `devices` and `accelerator` as CLI arguments ([#20913](https://github.com/Lightning-AI/pytorch-lightning/pull/20913))
|
|
21
|
+
- Raise ValueError when seed is `out-of-bounds` or `cannot be cast to int` ([#21029](https://github.com/Lightning-AI/pytorch-lightning/pull/21029))
|
|
22
|
+
|
|
23
|
+
### Fixed
|
|
24
|
+
|
|
25
|
+
- 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))
|
|
26
|
+
- Fixed remove extra `name` parameter in accelerator registry decorator ([#20975](https://github.com/Lightning-AI/pytorch-lightning/pull/20975))
|
|
27
|
+
|
|
28
|
+
|
|
9
29
|
## [2.5.2] - 2025-3-20
|
|
10
30
|
|
|
11
31
|
### 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
|
|