代码拉取完成，页面将自动刷新
提供模型下载链接及描述（可用表格）
| 模型 | 用途 | 下载链接 |
|---|---|---|
| Light_Clip | 检索 | 百度云(提取码:clip) |
| Light_VQGan_Dalle | 图像生成 | 百度云(提取码:dall) |
软件环境：pytorch1.8
硬件环境：NVIDIA V100
若需要使用多节点测试，可安装deepspeed框架：
pip install deepspeed
详细方法参见deepspeed官网
下载上述Light_CLIP模型文件model.pt下载至Light_CLIP文件夹下，运行以下脚本
python test.py
#output: Image probs: [[0.5521732 0.4478268]]
下载上述Light_VQGan_Dalle模型，将vqgan模型(提取码:iivw)下载至vqgan文件夹下，将bert-base-chinese(提取码:feur)、clip模型(同上述模型)下载至clip文件夹下。
运行：
cd Light_VQGan_Dalle
# single-gpu testing
./generate.sh <Light_VQGan_Dalle_model_path> <output_image_folder> <img_number> <generate_batch_size> <text_file_path>
# for example
./generate.sh ./dalle_small_model_pre_0921_lr4.5e-4_epoch6_data_point5762880.pt ./output_images 128 12 ./test_dalle.txt
# multi-gpu testing
./generate_multinodes.sh <hostfile_path> <Light_VQGan_Dalle_model_path> <output_image_folder> <img_number> <generate_batch_size> <text_file_path>
# for example
./generate_multinodes.sh ./hostfile.txt ./dalle_small_model_pre_0921_lr4.5e-4_epoch6_data_point5762880.pt ./output_images 128 12 ./test_dalle.txt
此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。
马建仓 AI 助手