|
@@ -29,7 +29,7 @@ def fix_pytorch_lightning():
|
|
if 'pytorch_lightning.utilities.distributed' not in sys.modules:
|
|
if 'pytorch_lightning.utilities.distributed' not in sys.modules:
|
|
import pytorch_lightning
|
|
import pytorch_lightning
|
|
# Lets the user know that the library was not found and then will set it to pytorch_lightning.utilities.rank_zero
|
|
# Lets the user know that the library was not found and then will set it to pytorch_lightning.utilities.rank_zero
|
|
- print(f"Pytorch_lightning.distributed not found, attempting pytorch_lightning.rank_zero")
|
|
|
|
|
|
+ print("Pytorch_lightning.distributed not found, attempting pytorch_lightning.rank_zero")
|
|
sys.modules["pytorch_lightning.utilities.distributed"] = pytorch_lightning.utilities.rank_zero
|
|
sys.modules["pytorch_lightning.utilities.distributed"] = pytorch_lightning.utilities.rank_zero
|
|
|
|
|
|
def fix_asyncio_event_loop_policy():
|
|
def fix_asyncio_event_loop_policy():
|