Updated requirements with pip install torch.distributed.checkpoint.format_utils

#2
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -16,6 +16,7 @@ pytorch-lightning==1.9.0
16
  safetensors
17
  timm==0.6.12
18
  torch
 
19
  torchvision
20
  transformers
21
  xformers
 
16
  safetensors
17
  timm==0.6.12
18
  torch
19
+ torch.distributed.checkpoint.format_utils
20
  torchvision
21
  transformers
22
  xformers