📜 蒸馏脚本:gen_urls.py
This commit is contained in:
parent
2852572958
commit
fa9671fa6c
14
scripts/distill/gen_urls.py
Normal file
14
scripts/distill/gen_urls.py
Normal file
@ -0,0 +1,14 @@
|
||||
#!/usr/bin/env python3
|
||||
"""生成COS预签名下载链接"""
|
||||
import os
|
||||
os.environ['ZY_OSS_KEY'] = 'AKIDkQuBQhoiS2OYXWebXLwMbdT7cvAScbbU'
|
||||
os.environ['ZY_OSS_SECRET'] = 'nPoZKArgUJBA4nJenjSxJSQBj5FCj3A4'
|
||||
from qcloud_cos import CosConfig, CosS3Client
|
||||
cfg = CosConfig(Region='ap-guangzhou', SecretId=os.environ['ZY_OSS_KEY'], SecretKey=os.environ['ZY_OSS_SECRET'])
|
||||
client = CosS3Client(cfg)
|
||||
bkt = 'sy-finetune-corpus-1317346199'
|
||||
for key in ['models/qwen25-15b-shuangyan-distill/model.safetensors',
|
||||
'models/qwen25-15b-coder-distill/model.safetensors']:
|
||||
url = client.get_presigned_url(Method='GET', Bucket=bkt, Key=key, Expired=604800)
|
||||
desc = key.split('/')[1].replace('-distill','')
|
||||
print(f'{desc}: {url}')
|
||||
Loading…
x
Reference in New Issue
Block a user