replacing fs for load_from_disk with storage_options

https://huggingface.co/docs/datasets/package_reference/loading_methods#datasets.load_from_disk
fs is deprecated now

scripts/train_unet.py

@@ -53,7 +53,7 @@ def main(args):
     if args.dataset_name is not None:
         if os.path.exists(args.dataset_name):
             dataset = load_from_disk(args.dataset_name,
-                                     args.dataset_config_name)["train"]
+                                     storage_options = args.dataset_config_name)["train"]
         else:
             dataset = load_dataset(
                 args.dataset_name,