Update README.md

Author: apoorvkh

README.md

@@ -70,10 +70,10 @@ We can distribute and run this function (e.g. on 2 machines x 2 GPUs) using **`t
 
 ```python
 import logging
-logging.basicConfig(level=logging.INFO)
-
 import torchrunx
 
+logging.basicConfig(level=logging.INFO)
+
 launcher = torchrunx.Launcher(
     hostnames = ["localhost", "second_machine"],  # or IP addresses
     workers_per_host = 2  # e.g. number of GPUs per host
@@ -93,7 +93,7 @@ trained_model: nn.Module = results.rank(0)
                      # or: results.index(hostname="localhost", local_rank=0)
 
 # and continue your script
-torch.save(trained_model.state_dict(), "output/model.pth")
+torch.save(trained_model.state_dict(), "outputs/model.pth")
 ```
 
 **See more examples where we fine-tune LLMs using:**