chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

clean up checkpoint build

Browse Source
This commit is contained in:
Nicholas Koben Kao
2024-01-24 22:20:00 -08:00
parent db04d02d34
commit 4c047aea62
3 changed files with 7 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files
builder/modal-builder/src/volume-builder/app.py
+1 -3
View File
@@ -43,7 +43,7 @@ def download_model(volume_name, download_config):
volume_base_path = vol_name_to_path[volume_name]
model_store_path = os.path.join(volume_base_path, folder_path)
modified_download_url = download_url + ("&" if "?" in download_url else "?") + "token=" + civitai_key
print('downlodaing', modified_download_url)
print('downloading', modified_download_url)
subprocess.run(["wget", modified_download_url , "--content-disposition", "-P", model_store_path])
subprocess.run(["ls", "-la", volume_base_path])
@@ -59,8 +59,6 @@ def download_model(volume_name, download_config):
@stub.local_entrypoint()
def simple_download():
import requests
print(vol_name_to_links)
print([(vol_name, link) for vol_name,link in vol_name_to_links.items()])
try:
list(download_model.starmap([(vol_name, link) for vol_name,link in vol_name_to_links.items()]))
except modal.exception.FunctionTimeoutError as e: