下载权重 - TRIO官方文档

在 WebUI 上复制权重ID：

将权重ID粘贴到下面代码的checkpoint_id处，执行脚本即可将Lora权重下载到本地。

import pytrio as trio
import requests
import os

client = trio.ServiceClient()
rest_client = client.create_rest_client()

checkpoint_id = "YOUR CHECKPOINT ID"
res = rest_client.get_archive_url(checkpoint_id)
print("Got the model download link::", res)

download_url = res["url"]
save_filename = f"{checkpoint_id}.zip"

with requests.get(download_url, stream=True) as response:
    response.raise_for_status()
    with open(save_filename, "wb") as f:
        for chunk in response.iter_content(chunk_size=8192):
            f.write(chunk)

print(f"File download complete! Save path: {os.path.abspath(save_filename)}")