Delete download.py

download.py

@@ -1,44 +0,0 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
-"""
-Functions for downloading pre-trained DiT models
-"""
-from torchvision.datasets.utils import download_url
-import torch
-import os
-
-
-
-def find_model(model_name):
-
-    checkpoint = torch.load(model_name, map_location=lambda storage, loc: storage)
-            
-    if "ema" in checkpoint:  # supports checkpoints from train.py
-            print('Ema existing!')
-            checkpoint = checkpoint["ema"]
-    return checkpoint
-
-
-def download_model(model_name):
-    """
-    Downloads a pre-trained DiT model from the web.
-    """
-    assert model_name in pretrained_models
-    local_path = f'pretrained_models/{model_name}'
-    if not os.path.isfile(local_path):
-        os.makedirs('pretrained_models', exist_ok=True)
-        web_path = f'https://dl.fbaipublicfiles.com/DiT/models/{model_name}'
-        download_url(web_path, 'pretrained_models')
-    model = torch.load(local_path, map_location=lambda storage, loc: storage)
-    return model
-
-
-if __name__ == "__main__":
-    # Download all DiT checkpoints
-    for model in pretrained_models:
-        download_model(model)
-    print('Done.')