第 1 步:确保环境变量已设置
在 PowerShell 中运行:
$env:HF_ENDPOINT="https://hf-mirror.com"
第 2 步:进入 Python
在 PowerShell 中运行:
python
在出现 >>> 后,复制下面这行代码运行。这行代码的作用是“告诉 Python 这个工具在哪里”:
from model import Kronos, KronosTokenizer, KronosPredictor
第 三步:下载模型副模型
现在运行这行代码,它会通过镜像站下载:
tokenizer = KronosTokenizer.from_pretrained("NeoQuasar/Kronos-Tokenizer-base")

 

加载主模型【下面命令任选其一】

 

model = Kronos.from_pretrained("NeoQuasar/Kronos-small") # 24.7M 参数,最快 model = Kronos.from_pretrained("NeoQuasar/Kronos-base") # 102M 参数

第四步:用自带数据快速上手

 

选退出python交互执行命令

exit()

项目里好像自带了HK_ali_09988_kline_5min_all.csv数据,我们需要把prediction_example.py代码的大概49行使用的数据改成‘HK_ali_09988_kline_5min_all.csv’数据, 并把D:\kronos\finetune_csv下的data 复制到D:\kronos\examples目录。 第49行代码如下:

df = pd.read_csv("./data/HK_ali_09988_kline_5min_all.csv")


运行模型



cd d:\kronos\examples
py -3 prediction_example.py

 

声明:本站内容来自公开平台,如若侵犯到您的权益,请联系我们,我们会第一时间删除!联系QQ:502428990。