浏览代码

await ensure_downloads_dir in brackets

Alex Cheema 1 年之前
父节点
当前提交
d17fdd223d
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      exo/download/new_shard_download.py

+ 2 - 2
exo/download/new_shard_download.py

@@ -50,7 +50,7 @@ async def ensure_downloads_dir() -> Path:
 
 async def delete_model(model_id: str, inference_engine_name: str) -> bool:
   repo_id = get_repo(model_id, inference_engine_name)
-  model_dir = await ensure_downloads_dir()/repo_id.replace("/", "--")
+  model_dir = (await ensure_downloads_dir())/repo_id.replace("/", "--")
   if not await aios.path.exists(model_dir): return False
   await asyncio.to_thread(shutil.rmtree, model_dir, ignore_errors=False)
   return True
@@ -199,7 +199,7 @@ async def download_shard(shard: Shard, inference_engine_classname: str, on_progr
   if DEBUG >= 2 and not skip_download: print(f"Downloading {shard.model_id=} for {inference_engine_classname}")
   repo_id = get_repo(shard.model_id, inference_engine_classname)
   revision = "main"
-  target_dir = await ensure_downloads_dir()/repo_id.replace("/", "--")
+  target_dir = (await ensure_downloads_dir())/repo_id.replace("/", "--")
   if not skip_download: await aios.makedirs(target_dir, exist_ok=True)
 
   if repo_id is None: