lightning 2.4.0.dev20240929__tar.gz → 2.5.0.dev20241027__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.
Files changed (298) hide show
  1. {lightning-2.4.0.dev20240929/src/lightning.egg-info → lightning-2.5.0.dev20241027}/PKG-INFO +11 -11
  2. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/fabric/strategies.txt +2 -1
  3. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/pytorch/extra.txt +2 -1
  4. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/bitsandbytes.py +8 -2
  5. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/deepspeed.py +1 -1
  6. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/xla_fsdp.py +1 -1
  7. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/progress/rich_progress.py +6 -31
  8. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/mlflow.py +1 -1
  9. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +1 -1
  10. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/deepspeed.py +1 -1
  11. lightning-2.5.0.dev20241027/src/lightning/version.info +1 -0
  12. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027/src/lightning.egg-info}/PKG-INFO +11 -11
  13. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning.egg-info/SOURCES.txt +1 -1
  14. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning.egg-info/requires.txt +60 -10
  15. lightning-2.5.0.dev20241027/src/version.info +1 -0
  16. lightning-2.4.0.dev20240929/src/lightning/version.info +0 -1
  17. lightning-2.4.0.dev20240929/src/version.info +0 -1
  18. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/.actions/assistant.py +0 -0
  19. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/CITATION.cff +0 -0
  20. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/LICENSE +0 -0
  21. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/README.md +0 -0
  22. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/pyproject.toml +0 -0
  23. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/base.txt +0 -0
  24. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/fabric/base.txt +0 -0
  25. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/fabric/docs.txt +0 -0
  26. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/fabric/examples.txt +0 -0
  27. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/fabric/test.txt +0 -0
  28. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/pytorch/base.txt +0 -0
  29. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/pytorch/docs.txt +0 -0
  30. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/pytorch/examples.txt +0 -0
  31. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/pytorch/strategies.txt +0 -0
  32. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/requirements/pytorch/test.txt +0 -0
  33. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/setup.cfg +0 -0
  34. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/setup.py +0 -0
  35. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/__about__.py +0 -0
  36. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/__init__.py +0 -0
  37. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/__setup__.py +0 -0
  38. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/__version__.py +0 -0
  39. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/data/__init__.py +0 -0
  40. /lightning-2.4.0.dev20240929/src/lightning/6354701a6af025b6f87424b50ec5812c5ba36bbb1e6e1b7a → /lightning-2.5.0.dev20241027/src/lightning/f8f2ff5651df291d54a9cfb34041c4a569caab36850b5d49 +0 -0
  41. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/CHANGELOG.md +0 -0
  42. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/__init__.py +0 -0
  43. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/_graveyard/__init__.py +0 -0
  44. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/_graveyard/tpu.py +0 -0
  45. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/accelerators/__init__.py +0 -0
  46. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/accelerators/accelerator.py +0 -0
  47. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/accelerators/cpu.py +0 -0
  48. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/accelerators/cuda.py +0 -0
  49. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/accelerators/mps.py +0 -0
  50. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/accelerators/registry.py +0 -0
  51. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/accelerators/xla.py +0 -0
  52. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/cli.py +0 -0
  53. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/connector.py +0 -0
  54. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/fabric.py +0 -0
  55. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/loggers/__init__.py +0 -0
  56. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/loggers/csv_logs.py +0 -0
  57. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/loggers/logger.py +0 -0
  58. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/loggers/tensorboard.py +0 -0
  59. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/__init__.py +0 -0
  60. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
  61. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
  62. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/collectives/single_device.py +0 -0
  63. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/collectives/torch_collective.py +0 -0
  64. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
  65. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
  66. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
  67. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/lightning.py +0 -0
  68. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/lsf.py +0 -0
  69. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
  70. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
  71. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
  72. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/environments/xla.py +0 -0
  73. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/io/__init__.py +0 -0
  74. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
  75. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/io/torch_io.py +0 -0
  76. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/io/xla.py +0 -0
  77. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
  78. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/amp.py +0 -0
  79. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/deepspeed.py +0 -0
  80. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/double.py +0 -0
  81. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/fsdp.py +0 -0
  82. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/half.py +0 -0
  83. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/precision.py +0 -0
  84. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/transformer_engine.py +0 -0
  85. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/utils.py +0 -0
  86. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/plugins/precision/xla.py +0 -0
  87. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/__init__.py +0 -0
  88. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/ddp.py +0 -0
  89. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/dp.py +0 -0
  90. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/fsdp.py +0 -0
  91. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
  92. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
  93. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/launchers/multiprocessing.py +0 -0
  94. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/launchers/subprocess_script.py +0 -0
  95. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
  96. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/model_parallel.py +0 -0
  97. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/parallel.py +0 -0
  98. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/registry.py +0 -0
  99. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/single_device.py +0 -0
  100. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/single_xla.py +0 -0
  101. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/strategy.py +0 -0
  102. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/strategies/xla.py +0 -0
  103. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/__init__.py +0 -0
  104. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/apply_func.py +0 -0
  105. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/cloud_io.py +0 -0
  106. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/consolidate_checkpoint.py +0 -0
  107. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/data.py +0 -0
  108. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/device_dtype_mixin.py +0 -0
  109. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/device_parser.py +0 -0
  110. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/distributed.py +0 -0
  111. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/enums.py +0 -0
  112. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/exceptions.py +0 -0
  113. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/imports.py +0 -0
  114. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/init.py +0 -0
  115. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/load.py +0 -0
  116. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/logger.py +0 -0
  117. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/optimizer.py +0 -0
  118. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/rank_zero.py +0 -0
  119. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/registry.py +0 -0
  120. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/seed.py +0 -0
  121. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/spike.py +0 -0
  122. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
  123. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/testing/_runif.py +0 -0
  124. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/throughput.py +0 -0
  125. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/types.py +0 -0
  126. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/utilities/warnings.py +0 -0
  127. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/fabric/wrappers.py +0 -0
  128. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/py.typed +0 -0
  129. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/CHANGELOG.md +0 -0
  130. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/__init__.py +0 -0
  131. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
  132. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
  133. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
  134. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/_graveyard/precision.py +0 -0
  135. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
  136. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/accelerators/__init__.py +0 -0
  137. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
  138. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/accelerators/cpu.py +0 -0
  139. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/accelerators/cuda.py +0 -0
  140. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/accelerators/mps.py +0 -0
  141. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/accelerators/xla.py +0 -0
  142. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/__init__.py +0 -0
  143. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
  144. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/callback.py +0 -0
  145. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
  146. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/device_stats_monitor.py +0 -0
  147. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/early_stopping.py +0 -0
  148. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/finetuning.py +0 -0
  149. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +0 -0
  150. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
  151. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/lr_finder.py +0 -0
  152. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/lr_monitor.py +0 -0
  153. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/model_checkpoint.py +0 -0
  154. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/model_summary.py +0 -0
  155. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
  156. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/prediction_writer.py +0 -0
  157. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
  158. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/progress/progress_bar.py +0 -0
  159. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +0 -0
  160. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/pruning.py +0 -0
  161. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/rich_model_summary.py +0 -0
  162. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/spike.py +0 -0
  163. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +0 -0
  164. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/throughput_monitor.py +0 -0
  165. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/callbacks/timer.py +0 -0
  166. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/cli.py +0 -0
  167. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/__init__.py +0 -0
  168. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/datamodule.py +0 -0
  169. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/hooks.py +0 -0
  170. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
  171. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/mixins/hparams_mixin.py +0 -0
  172. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/module.py +0 -0
  173. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/optimizer.py +0 -0
  174. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/core/saving.py +0 -0
  175. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/demos/__init__.py +0 -0
  176. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/demos/boring_classes.py +0 -0
  177. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/demos/lstm.py +0 -0
  178. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
  179. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/demos/transformer.py +0 -0
  180. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/__init__.py +0 -0
  181. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/comet.py +0 -0
  182. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/csv_logs.py +0 -0
  183. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/logger.py +0 -0
  184. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/neptune.py +0 -0
  185. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/tensorboard.py +0 -0
  186. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/utilities.py +0 -0
  187. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loggers/wandb.py +0 -0
  188. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/__init__.py +0 -0
  189. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/evaluation_loop.py +0 -0
  190. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/fetchers.py +0 -0
  191. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/fit_loop.py +0 -0
  192. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/loop.py +0 -0
  193. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
  194. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/optimization/automatic.py +0 -0
  195. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
  196. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/optimization/manual.py +0 -0
  197. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/prediction_loop.py +0 -0
  198. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/progress.py +0 -0
  199. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/training_epoch_loop.py +0 -0
  200. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/loops/utilities.py +0 -0
  201. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/overrides/__init__.py +0 -0
  202. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/overrides/distributed.py +0 -0
  203. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/__init__.py +0 -0
  204. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
  205. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
  206. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/io/async_plugin.py +0 -0
  207. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
  208. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
  209. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/io/wrapper.py +0 -0
  210. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
  211. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
  212. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
  213. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/amp.py +0 -0
  214. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/deepspeed.py +0 -0
  215. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/double.py +0 -0
  216. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/fsdp.py +0 -0
  217. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/half.py +0 -0
  218. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/precision.py +0 -0
  219. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
  220. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
  221. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/profilers/__init__.py +0 -0
  222. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/profilers/advanced.py +0 -0
  223. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/profilers/base.py +0 -0
  224. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/profilers/profiler.py +0 -0
  225. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/profilers/pytorch.py +0 -0
  226. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/profilers/simple.py +0 -0
  227. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/profilers/xla.py +0 -0
  228. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/serve/__init__.py +0 -0
  229. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/serve/servable_module.py +0 -0
  230. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/serve/servable_module_validator.py +0 -0
  231. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/__init__.py +0 -0
  232. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/ddp.py +0 -0
  233. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/fsdp.py +0 -0
  234. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
  235. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
  236. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +0 -0
  237. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +0 -0
  238. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
  239. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/model_parallel.py +0 -0
  240. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/parallel.py +0 -0
  241. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/single_device.py +0 -0
  242. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/single_xla.py +0 -0
  243. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/strategy.py +0 -0
  244. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/strategies/xla.py +0 -0
  245. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/__init__.py +0 -0
  246. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/call.py +0 -0
  247. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
  248. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
  249. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +0 -0
  250. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/callback_connector.py +0 -0
  251. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +0 -0
  252. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/data_connector.py +0 -0
  253. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
  254. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
  255. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +0 -0
  256. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +0 -0
  257. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/connectors/signal_connector.py +0 -0
  258. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/setup.py +0 -0
  259. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/states.py +0 -0
  260. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/trainer/trainer.py +0 -0
  261. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/tuner/__init__.py +0 -0
  262. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/tuner/batch_size_scaling.py +0 -0
  263. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/tuner/lr_finder.py +0 -0
  264. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/tuner/tuning.py +0 -0
  265. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/__init__.py +0 -0
  266. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/_pytree.py +0 -0
  267. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/argparse.py +0 -0
  268. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/combined_loader.py +0 -0
  269. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/compile.py +0 -0
  270. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +0 -0
  271. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/data.py +0 -0
  272. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/deepspeed.py +0 -0
  273. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/enums.py +0 -0
  274. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/exceptions.py +0 -0
  275. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/grads.py +0 -0
  276. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/imports.py +0 -0
  277. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/memory.py +0 -0
  278. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
  279. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
  280. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/migration/utils.py +0 -0
  281. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/model_helpers.py +0 -0
  282. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
  283. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/model_summary/model_summary.py +0 -0
  284. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +0 -0
  285. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
  286. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/parsing.py +0 -0
  287. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
  288. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/seed.py +0 -0
  289. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
  290. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
  291. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/testing/_runif.py +0 -0
  292. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/types.py +0 -0
  293. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +0 -0
  294. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning/pytorch/utilities/warnings.py +0 -0
  295. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning.egg-info/dependency_links.txt +0 -0
  296. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning.egg-info/entry_points.txt +0 -0
  297. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning.egg-info/not-zip-safe +0 -0
  298. {lightning-2.4.0.dev20240929 → lightning-2.5.0.dev20241027}/src/lightning.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: lightning
3
- Version: 2.4.0.dev20240929
3
+ Version: 2.5.0.dev20241027
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
  Author: Lightning AI et al.
@@ -27,22 +27,22 @@ Classifier: Programming Language :: Python :: 3.11
27
27
  Classifier: Programming Language :: Python :: 3.12
28
28
  Requires-Python: >=3.9
29
29
  Description-Content-Type: text/markdown
30
- Provides-Extra: pytorch-examples
30
+ Provides-Extra: fabric-test
31
+ Provides-Extra: fabric-strategies
32
+ Provides-Extra: fabric-examples
33
+ Provides-Extra: pytorch-test
31
34
  Provides-Extra: pytorch-strategies
32
35
  Provides-Extra: pytorch-extra
33
- Provides-Extra: pytorch-test
34
- Provides-Extra: fabric-examples
35
- Provides-Extra: fabric-strategies
36
- Provides-Extra: fabric-test
36
+ Provides-Extra: pytorch-examples
37
37
  Provides-Extra: fabric-all
38
38
  Provides-Extra: fabric-dev
39
39
  Provides-Extra: pytorch-all
40
40
  Provides-Extra: pytorch-dev
41
- Provides-Extra: examples
42
- Provides-Extra: strategies
43
- Provides-Extra: extra
44
41
  Provides-Extra: test
42
+ Provides-Extra: strategies
43
+ Provides-Extra: examples
45
44
  Provides-Extra: data
45
+ Provides-Extra: extra
46
46
  Provides-Extra: all
47
47
  Provides-Extra: dev
48
48
  License-File: LICENSE
@@ -67,7 +67,7 @@ ______________________________________________________________________
67
67
  <a href="#lightning-fabric-expert-control">Fabric</a> •
68
68
  <a href="https://lightning.ai/">Lightning AI</a> •
69
69
  <a href="#community">Community</a> •
70
- <a href="https://pytorch-lightning.readthedocs.io/en/2.4.0.dev20240929">Docs</a>
70
+ <a href="https://pytorch-lightning.readthedocs.io/en/2.5.0.dev20241027">Docs</a>
71
71
  </p>
72
72
 
73
73
  <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -592,7 +592,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
592
592
 
593
593
  | System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
594
594
  | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
595
- | Linux py3.9 \[GPUs\] | | | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.4.0.dev20240929)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
595
+ | Linux py3.9 \[GPUs\] | | | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.5.0.dev20241027)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
596
596
  | Linux py3.9 \[TPUs\] | | [![Test PyTorch - TPU](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml) | |
597
597
  | Linux (multiple Python versions) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
598
598
  | OSX (multiple Python versions) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
@@ -6,4 +6,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
8
  deepspeed >=0.8.2, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
9
- bitsandbytes >=0.42.0,<0.43.0
9
+ bitsandbytes >=0.44.0,<0.44.2; sys_platform == 'linux' or sys_platform == 'win32'
10
+ bitsandbytes >=0.42.0,<0.43.0 ; sys_platform == 'darwin'
@@ -8,4 +8,5 @@ hydra-core >=1.2.0, <1.4.0
8
8
  jsonargparse[signatures] >=4.27.7, <4.28.0
9
9
  rich >=12.3.0, <13.6.0
10
10
  tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute
11
- bitsandbytes >=0.42.0,<0.43.0
11
+ bitsandbytes >=0.44.0,<0.44.2; sys_platform == 'linux' or sys_platform == 'win32'
12
+ bitsandbytes >=0.42.0,<0.43.0 ; sys_platform == 'darwin'
@@ -43,7 +43,7 @@ _BITSANDBYTES_AVAILABLE = RequirementCache("bitsandbytes>=0.42.0")
43
43
 
44
44
 
45
45
  class BitsandbytesPrecision(Precision):
46
- """Plugin for quantizing weights with `bitsandbytes <https://github.com/TimDettmers/bitsandbytes>`__.
46
+ """Plugin for quantizing weights with `bitsandbytes <https://github.com/bitsandbytes-foundation/bitsandbytes>`__.
47
47
 
48
48
  .. warning:: This is an :ref:`experimental <versioning:Experimental API>` feature.
49
49
 
@@ -184,11 +184,15 @@ def _replace_param(
184
184
  if param.device.type == "meta":
185
185
  if isinstance(param, bnb.nn.Params4bit):
186
186
  return bnb.nn.Params4bit(
187
- data,
187
+ data=data,
188
188
  requires_grad=data.requires_grad,
189
189
  quant_state=quant_state,
190
+ blocksize=param.blocksize,
190
191
  compress_statistics=param.compress_statistics,
191
192
  quant_type=param.quant_type,
193
+ quant_storage=param.quant_storage,
194
+ module=param.module,
195
+ bnb_quantized=param.bnb_quantized,
192
196
  )
193
197
  return torch.nn.Parameter(data, requires_grad=data.requires_grad)
194
198
  param.data = data
@@ -322,6 +326,7 @@ def _import_bitsandbytes() -> ModuleType:
322
326
  return
323
327
  assert isinstance(self.weight, bnb.nn.Params4bit)
324
328
  self.weight = self.quantize(self.weight, weight, device)
329
+ self.weight.bnb_quantized = True
325
330
 
326
331
  @staticmethod
327
332
  def quantize(
@@ -337,6 +342,7 @@ def _import_bitsandbytes() -> ModuleType:
337
342
  blocksize=params4bit.blocksize,
338
343
  compress_statistics=params4bit.compress_statistics,
339
344
  quant_type=params4bit.quant_type,
345
+ quant_storage=params4bit.quant_storage,
340
346
  )
341
347
  return _replace_param(params4bit, w_4bit, quant_state)
342
348
 
@@ -598,7 +598,7 @@ class DeepSpeedStrategy(DDPStrategy, _Sharded):
598
598
  ) -> Tuple["DeepSpeedEngine", Optimizer]:
599
599
  """Initialize one model and one optimizer with an optional learning rate scheduler.
600
600
 
601
- This calls :func:`deepspeed.initialize` internally.
601
+ This calls ``deepspeed.initialize`` internally.
602
602
 
603
603
  """
604
604
  import deepspeed
@@ -56,7 +56,7 @@ class XLAFSDPStrategy(ParallelStrategy, _Sharded):
56
56
 
57
57
  .. warning:: This is an :ref:`experimental <versioning:Experimental API>` feature.
58
58
 
59
- For more information check out https://github.com/pytorch/xla/blob/master/docs/fsdp.md
59
+ For more information check out https://github.com/pytorch/xla/blob/v2.5.0/docs/fsdp.md
60
60
 
61
61
  Args:
62
62
  auto_wrap_policy: Same as ``auto_wrap_policy`` parameter in
@@ -206,14 +206,14 @@ class RichProgressBarTheme:
206
206
 
207
207
  """
208
208
 
209
- description: Union[str, "Style"] = "white"
209
+ description: Union[str, "Style"] = ""
210
210
  progress_bar: Union[str, "Style"] = "#6206E0"
211
211
  progress_bar_finished: Union[str, "Style"] = "#6206E0"
212
212
  progress_bar_pulse: Union[str, "Style"] = "#6206E0"
213
- batch_progress: Union[str, "Style"] = "white"
214
- time: Union[str, "Style"] = "grey54"
215
- processing_speed: Union[str, "Style"] = "grey70"
216
- metrics: Union[str, "Style"] = "white"
213
+ batch_progress: Union[str, "Style"] = ""
214
+ time: Union[str, "Style"] = "dim"
215
+ processing_speed: Union[str, "Style"] = "dim underline"
216
+ metrics: Union[str, "Style"] = "italic"
217
217
  metrics_text_delimiter: str = " "
218
218
  metrics_format: str = ".3f"
219
219
 
@@ -280,7 +280,6 @@ class RichProgressBar(ProgressBar):
280
280
  self._metric_component: Optional[MetricsTextColumn] = None
281
281
  self._progress_stopped: bool = False
282
282
  self.theme = theme
283
- self._update_for_light_colab_theme()
284
283
 
285
284
  @property
286
285
  def refresh_rate(self) -> float:
@@ -318,13 +317,6 @@ class RichProgressBar(ProgressBar):
318
317
  assert self.test_progress_bar_id is not None
319
318
  return self.progress.tasks[self.test_progress_bar_id]
320
319
 
321
- def _update_for_light_colab_theme(self) -> None:
322
- if _detect_light_colab_theme():
323
- attributes = ["description", "batch_progress", "metrics"]
324
- for attr in attributes:
325
- if getattr(self.theme, attr) == "white":
326
- setattr(self.theme, attr, "black")
327
-
328
320
  @override
329
321
  def disable(self) -> None:
330
322
  self._enabled = False
@@ -449,7 +441,7 @@ class RichProgressBar(ProgressBar):
449
441
  def _add_task(self, total_batches: Union[int, float], description: str, visible: bool = True) -> "TaskID":
450
442
  assert self.progress is not None
451
443
  return self.progress.add_task(
452
- f"[{self.theme.description}]{description}",
444
+ f"[{self.theme.description}]{description}" if self.theme.description else description,
453
445
  total=total_batches,
454
446
  visible=visible,
455
447
  )
@@ -656,20 +648,3 @@ class RichProgressBar(ProgressBar):
656
648
  state["progress"] = None
657
649
  state["_console"] = None
658
650
  return state
659
-
660
-
661
- def _detect_light_colab_theme() -> bool:
662
- """Detect if it's light theme in Colab."""
663
- try:
664
- import get_ipython
665
- except (NameError, ModuleNotFoundError):
666
- return False
667
- ipython = get_ipython()
668
- if "google.colab" in str(ipython.__class__):
669
- try:
670
- from google.colab import output
671
-
672
- return output.eval_js('document.documentElement.matches("[theme=light]")')
673
- except ModuleNotFoundError:
674
- return False
675
- return False
@@ -360,7 +360,7 @@ class MLFlowLogger(Logger):
360
360
  aliases = ["latest", "best"] if p == checkpoint_callback.best_model_path else ["latest"]
361
361
 
362
362
  # Artifact path on mlflow
363
- artifact_path = f"model/checkpoints/{Path(p).stem}"
363
+ artifact_path = Path(p).stem
364
364
 
365
365
  # Log the checkpoint
366
366
  self.experiment.log_artifact(self._run_id, p, artifact_path)
@@ -16,7 +16,7 @@ from lightning.pytorch.plugins.precision.precision import Precision
16
16
 
17
17
 
18
18
  class BitsandbytesPrecision(Precision, FabricBNBPrecision):
19
- """Plugin for quantizing weights with `bitsandbytes <https://github.com/TimDettmers/bitsandbytes>`__.
19
+ """Plugin for quantizing weights with `bitsandbytes <https://github.com/bitsandbytes-foundation/bitsandbytes>`__.
20
20
 
21
21
  .. warning:: This is an :ref:`experimental <versioning:Experimental API>` feature.
22
22
 
@@ -414,7 +414,7 @@ class DeepSpeedStrategy(DDPStrategy):
414
414
  ) -> Tuple["deepspeed.DeepSpeedEngine", Optimizer]:
415
415
  """Initialize one model and one optimizer with an optional learning rate scheduler.
416
416
 
417
- This calls :func:`deepspeed.initialize` internally.
417
+ This calls ``deepspeed.initialize`` internally.
418
418
 
419
419
  """
420
420
  import deepspeed
@@ -0,0 +1 @@
1
+ 2.5.0.dev20241027
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: lightning
3
- Version: 2.4.0.dev20240929
3
+ Version: 2.5.0.dev20241027
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
  Author: Lightning AI et al.
@@ -27,22 +27,22 @@ Classifier: Programming Language :: Python :: 3.11
27
27
  Classifier: Programming Language :: Python :: 3.12
28
28
  Requires-Python: >=3.9
29
29
  Description-Content-Type: text/markdown
30
- Provides-Extra: pytorch-examples
30
+ Provides-Extra: fabric-test
31
+ Provides-Extra: fabric-strategies
32
+ Provides-Extra: fabric-examples
33
+ Provides-Extra: pytorch-test
31
34
  Provides-Extra: pytorch-strategies
32
35
  Provides-Extra: pytorch-extra
33
- Provides-Extra: pytorch-test
34
- Provides-Extra: fabric-examples
35
- Provides-Extra: fabric-strategies
36
- Provides-Extra: fabric-test
36
+ Provides-Extra: pytorch-examples
37
37
  Provides-Extra: fabric-all
38
38
  Provides-Extra: fabric-dev
39
39
  Provides-Extra: pytorch-all
40
40
  Provides-Extra: pytorch-dev
41
- Provides-Extra: examples
42
- Provides-Extra: strategies
43
- Provides-Extra: extra
44
41
  Provides-Extra: test
42
+ Provides-Extra: strategies
43
+ Provides-Extra: examples
45
44
  Provides-Extra: data
45
+ Provides-Extra: extra
46
46
  Provides-Extra: all
47
47
  Provides-Extra: dev
48
48
  License-File: LICENSE
@@ -67,7 +67,7 @@ ______________________________________________________________________
67
67
  <a href="#lightning-fabric-expert-control">Fabric</a> •
68
68
  <a href="https://lightning.ai/">Lightning AI</a> •
69
69
  <a href="#community">Community</a> •
70
- <a href="https://pytorch-lightning.readthedocs.io/en/2.4.0.dev20240929">Docs</a>
70
+ <a href="https://pytorch-lightning.readthedocs.io/en/2.5.0.dev20241027">Docs</a>
71
71
  </p>
72
72
 
73
73
  <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -592,7 +592,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
592
592
 
593
593
  | System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
594
594
  | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
595
- | Linux py3.9 \[GPUs\] | | | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.4.0.dev20240929)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
595
+ | Linux py3.9 \[GPUs\] | | | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.5.0.dev20241027)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
596
596
  | Linux py3.9 \[TPUs\] | | [![Test PyTorch - TPU](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml) | |
597
597
  | Linux (multiple Python versions) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
598
598
  | OSX (multiple Python versions) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
@@ -17,11 +17,11 @@ requirements/pytorch/extra.txt
17
17
  requirements/pytorch/strategies.txt
18
18
  requirements/pytorch/test.txt
19
19
  src/version.info
20
- src/lightning/6354701a6af025b6f87424b50ec5812c5ba36bbb1e6e1b7a
21
20
  src/lightning/__about__.py
22
21
  src/lightning/__init__.py
23
22
  src/lightning/__setup__.py
24
23
  src/lightning/__version__.py
24
+ src/lightning/f8f2ff5651df291d54a9cfb34041c4a569caab36850b5d49
25
25
  src/lightning/py.typed
26
26
  src/lightning/version.info
27
27
  src/lightning.egg-info/PKG-INFO
@@ -9,7 +9,6 @@ typing-extensions<6.0,>=4.4.0
9
9
  pytorch-lightning
10
10
 
11
11
  [all]
12
- bitsandbytes<1.0,>=0.42.0
13
12
  hydra-core<2.0,>=1.2.0
14
13
  ipython[all]<9.0
15
14
  jsonargparse[signatures]<5.0,>=4.27.7
@@ -25,11 +24,16 @@ torchvision<1.0,>=0.16.0
25
24
  [all:platform_system != "Windows" and platform_system != "Darwin"]
26
25
  deepspeed<=0.9.3,>=0.8.2
27
26
 
27
+ [all:sys_platform == "darwin"]
28
+ bitsandbytes<1.0,>=0.42.0
29
+
30
+ [all:sys_platform == "linux" or sys_platform == "win32"]
31
+ bitsandbytes<1.0,>=0.44.0
32
+
28
33
  [data]
29
34
  litdata<1.0,>=0.2.0rc
30
35
 
31
36
  [dev]
32
- bitsandbytes<1.0,>=0.42.0
33
37
  click==8.1.7
34
38
  cloudpickle<3.0,>=1.3
35
39
  coverage==7.3.1
@@ -63,6 +67,12 @@ uvicorn
63
67
  [dev:platform_system != "Windows" and platform_system != "Darwin"]
64
68
  deepspeed<=0.9.3,>=0.8.2
65
69
 
70
+ [dev:sys_platform == "darwin"]
71
+ bitsandbytes<1.0,>=0.42.0
72
+
73
+ [dev:sys_platform == "linux" or sys_platform == "win32"]
74
+ bitsandbytes<1.0,>=0.44.0
75
+
66
76
  [examples]
67
77
  ipython[all]<9.0
68
78
  lightning-utilities<1.0,>=0.8.0
@@ -71,7 +81,6 @@ torchmetrics<2.0,>=0.10.0
71
81
  torchvision<1.0,>=0.16.0
72
82
 
73
83
  [extra]
74
- bitsandbytes<1.0,>=0.42.0
75
84
  hydra-core<2.0,>=1.2.0
76
85
  jsonargparse[signatures]<5.0,>=4.27.7
77
86
  matplotlib<4.0,>3.1
@@ -79,8 +88,13 @@ omegaconf<3.0,>=2.2.3
79
88
  rich<14.0,>=12.3.0
80
89
  tensorboardX<3.0,>=2.2
81
90
 
82
- [fabric-all]
91
+ [extra:sys_platform == "darwin"]
83
92
  bitsandbytes<1.0,>=0.42.0
93
+
94
+ [extra:sys_platform == "linux" or sys_platform == "win32"]
95
+ bitsandbytes<1.0,>=0.44.0
96
+
97
+ [fabric-all]
84
98
  lightning-utilities<1.0,>=0.8.0
85
99
  torchmetrics<2.0,>=0.10.0
86
100
  torchvision<1.0,>=0.16.0
@@ -88,8 +102,13 @@ torchvision<1.0,>=0.16.0
88
102
  [fabric-all:platform_system != "Windows" and platform_system != "Darwin"]
89
103
  deepspeed<=0.9.3,>=0.8.2
90
104
 
91
- [fabric-dev]
105
+ [fabric-all:sys_platform == "darwin"]
92
106
  bitsandbytes<1.0,>=0.42.0
107
+
108
+ [fabric-all:sys_platform == "linux" or sys_platform == "win32"]
109
+ bitsandbytes<1.0,>=0.44.0
110
+
111
+ [fabric-dev]
93
112
  click==8.1.7
94
113
  coverage==7.3.1
95
114
  lightning-utilities<1.0,>=0.8.0
@@ -107,17 +126,28 @@ torchvision<1.0,>=0.16.0
107
126
  [fabric-dev:platform_system != "Windows" and platform_system != "Darwin"]
108
127
  deepspeed<=0.9.3,>=0.8.2
109
128
 
129
+ [fabric-dev:sys_platform == "darwin"]
130
+ bitsandbytes<1.0,>=0.42.0
131
+
132
+ [fabric-dev:sys_platform == "linux" or sys_platform == "win32"]
133
+ bitsandbytes<1.0,>=0.44.0
134
+
110
135
  [fabric-examples]
111
136
  lightning-utilities<1.0,>=0.8.0
112
137
  torchmetrics<2.0,>=0.10.0
113
138
  torchvision<1.0,>=0.16.0
114
139
 
115
140
  [fabric-strategies]
116
- bitsandbytes<1.0,>=0.42.0
117
141
 
118
142
  [fabric-strategies:platform_system != "Windows" and platform_system != "Darwin"]
119
143
  deepspeed<=0.9.3,>=0.8.2
120
144
 
145
+ [fabric-strategies:sys_platform == "darwin"]
146
+ bitsandbytes<1.0,>=0.42.0
147
+
148
+ [fabric-strategies:sys_platform == "linux" or sys_platform == "win32"]
149
+ bitsandbytes<1.0,>=0.44.0
150
+
121
151
  [fabric-test]
122
152
  click==8.1.7
123
153
  coverage==7.3.1
@@ -131,7 +161,6 @@ tensorboardX<3.0,>=2.2
131
161
  torchmetrics<2.0,>=0.7.0
132
162
 
133
163
  [pytorch-all]
134
- bitsandbytes<1.0,>=0.42.0
135
164
  hydra-core<2.0,>=1.2.0
136
165
  ipython[all]<9.0
137
166
  jsonargparse[signatures]<5.0,>=4.27.7
@@ -147,8 +176,13 @@ torchvision<1.0,>=0.16.0
147
176
  [pytorch-all:platform_system != "Windows" and platform_system != "Darwin"]
148
177
  deepspeed<=0.9.3,>=0.8.2
149
178
 
150
- [pytorch-dev]
179
+ [pytorch-all:sys_platform == "darwin"]
151
180
  bitsandbytes<1.0,>=0.42.0
181
+
182
+ [pytorch-all:sys_platform == "linux" or sys_platform == "win32"]
183
+ bitsandbytes<1.0,>=0.44.0
184
+
185
+ [pytorch-dev]
152
186
  cloudpickle<3.0,>=1.3
153
187
  coverage==7.3.1
154
188
  fastapi
@@ -180,6 +214,12 @@ uvicorn
180
214
  [pytorch-dev:platform_system != "Windows" and platform_system != "Darwin"]
181
215
  deepspeed<=0.9.3,>=0.8.2
182
216
 
217
+ [pytorch-dev:sys_platform == "darwin"]
218
+ bitsandbytes<1.0,>=0.42.0
219
+
220
+ [pytorch-dev:sys_platform == "linux" or sys_platform == "win32"]
221
+ bitsandbytes<1.0,>=0.44.0
222
+
183
223
  [pytorch-examples]
184
224
  ipython[all]<9.0
185
225
  lightning-utilities<1.0,>=0.8.0
@@ -188,7 +228,6 @@ torchmetrics<2.0,>=0.10.0
188
228
  torchvision<1.0,>=0.16.0
189
229
 
190
230
  [pytorch-extra]
191
- bitsandbytes<1.0,>=0.42.0
192
231
  hydra-core<2.0,>=1.2.0
193
232
  jsonargparse[signatures]<5.0,>=4.27.7
194
233
  matplotlib<4.0,>3.1
@@ -196,6 +235,12 @@ omegaconf<3.0,>=2.2.3
196
235
  rich<14.0,>=12.3.0
197
236
  tensorboardX<3.0,>=2.2
198
237
 
238
+ [pytorch-extra:sys_platform == "darwin"]
239
+ bitsandbytes<1.0,>=0.42.0
240
+
241
+ [pytorch-extra:sys_platform == "linux" or sys_platform == "win32"]
242
+ bitsandbytes<1.0,>=0.44.0
243
+
199
244
  [pytorch-strategies]
200
245
 
201
246
  [pytorch-strategies:platform_system != "Windows" and platform_system != "Darwin"]
@@ -220,11 +265,16 @@ tensorboard<3.0,>=2.9.1
220
265
  uvicorn
221
266
 
222
267
  [strategies]
223
- bitsandbytes<1.0,>=0.42.0
224
268
 
225
269
  [strategies:platform_system != "Windows" and platform_system != "Darwin"]
226
270
  deepspeed<=0.9.3,>=0.8.2
227
271
 
272
+ [strategies:sys_platform == "darwin"]
273
+ bitsandbytes<1.0,>=0.42.0
274
+
275
+ [strategies:sys_platform == "linux" or sys_platform == "win32"]
276
+ bitsandbytes<1.0,>=0.44.0
277
+
228
278
  [test]
229
279
  click==8.1.7
230
280
  cloudpickle<3.0,>=1.3
@@ -0,0 +1 @@
1
+ 2.5.0.dev20241027
@@ -1 +0,0 @@
1
- 2.4.0.dev20240929
@@ -1 +0,0 @@
1
- 2.4.0.dev20240929