From 8cf57b3638e13c0d5776d4bfc417f3e634aa55d1 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Fri, 23 Feb 2024 11:58:15 +0000 Subject: [PATCH 1/2] update --- scripts/convert_dance_diffusion_to_diffusers.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/scripts/convert_dance_diffusion_to_diffusers.py b/scripts/convert_dance_diffusion_to_diffusers.py index d53d1f792e89..c56fb8bfacf9 100755 --- a/scripts/convert_dance_diffusion_to_diffusers.py +++ b/scripts/convert_dance_diffusion_to_diffusers.py @@ -4,6 +4,7 @@ import os from copy import deepcopy +import requests import torch from audio_diffusion.models import DiffusionAttnUnet1D from diffusion import sampling @@ -73,9 +74,14 @@ def __init__(self, global_args): def download(model_name): url = MODELS_MAP[model_name]["url"] - os.system(f"wget {url} ./") + r = requests.get(url, stream=True) - return f"./{model_name}.ckpt" + local_filename = f"./{model_name}.ckpt" + with open(local_filename, 'wb') as fp: + for chunk in r.iter_content(chunk_size=8192): + fp.write(chunk) + + return local_filename DOWN_NUM_TO_LAYER = { From 53bc88e1d86576364e4034752cbd5a94d499a6e5 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Mon, 26 Feb 2024 03:49:17 +0000 Subject: [PATCH 2/2] make style --- scripts/convert_dance_diffusion_to_diffusers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/convert_dance_diffusion_to_diffusers.py b/scripts/convert_dance_diffusion_to_diffusers.py index c56fb8bfacf9..ce69bfe2bfc8 100755 --- a/scripts/convert_dance_diffusion_to_diffusers.py +++ b/scripts/convert_dance_diffusion_to_diffusers.py @@ -77,7 +77,7 @@ def download(model_name): r = requests.get(url, stream=True) local_filename = f"./{model_name}.ckpt" - with open(local_filename, 'wb') as fp: + with open(local_filename, "wb") as fp: for chunk in r.iter_content(chunk_size=8192): fp.write(chunk)