bugfix: distribute rank initialization

[Squire-tomsk]

Change RANK env variable value initialization from local_gpu_id to rank.

This change is necessary because in the current configuration, the get_rank function returns the GPU id get_rank . As a result, the loggers fail to initialize properly, causing issues here.

[CLAassistant]

All committers have signed the CLA.