NewBie-image-Exp0.1 是一个预置好的动漫图像生成镜像,基于 Next-DiT 架构,内置了参数规模 3.5B(35 亿)的模型。它最大的好处是省去了环境搭建和权重下载的折腾——Python 依赖、CUDA 兼容性、源码 bug 都已经在镜像里处理完毕。目前推理大约占用 14-15 GB 显存,所以建议至少 16 GB。
上手:生成第一张图
进入容器后,先切到镜像主目录:
cd ..
cd NewBie-image-Exp0.1
然后直接运行预置的测试脚本:
python test.py
模型加载后,几十秒内就会在当前目录生成一张 success_output.png,是默认的动漫风格人物。如果没看到文件,多半是显存不足,终端会有 CUDA out of memory 报错。
用 XML 提示词精确控制角色
test.py 里的 prompt 变量就是控制生成内容的入口。它采用 XML 标签,能分别指定每个角色的名称、性别、外观,以及整体风格。一个基本的例子:
prompt = """ <character_1> <n>miku</n> <gender>1girl</gender> <appearance>blue_hair, long_twintails, teal_eyes</appearance> </character_1> <general_tags> <style>anime_style, high_quality</style> </general_tags> """
<character_1> 定义第一个角色,可扩展到 <character_2>、<character_3>;<n> 填写角色名或模板;<gender> 常用 1girl、1boy 之类;<appearance> 用逗号分隔多项外貌特征;<general_tags> 里的 style 影响整体画质与画风。
改成红发短发的女孩也非常简单:
prompt = """ <character_1> <n>original</n> <gender>1girl</gender> <appearance>red_hair, short_hair, green_eyes, smiling</appearance> </character_1> <general_tags> <style>anime_style, ultra_detailed, best_quality</style> </general_tags> """
保存后重新运行 python test.py,生成结果就会跟着变化。
XML 结构在多角色场景里优势更明显,因为每个角色的属性是独立声明的。比如一男一女:
prompt = """ <character_1> <n>original</n> <gender>1girl</gender> <appearance>blue_hair, long_twintails, blue_eyes, school_uniform</appearance> </character_1> <character_2> <n>original</n> <gender>1boy</gender> <appearance>black_hair, short_hair, brown_eyes, casual_jacket</appearance> </character_2> <general_tags> <style>anime_style, dynamic_pose, background_scenery</style> </general_tags> """
这样,模型就能分别理解两个角色的设定,安排在同一画面里。
切换交互模式:create.py
如果每次都要修改文件比较麻烦,镜像里还提供了 create.py,一个交互式的生成脚本:
python create.py
运行后直接输入 XML 提示词,回车就能生成图像,然后问你要不要继续,适合一口气测试很多想法。
镜像内的文件组织大致是:
test.py:基础推理脚本,方便快速测试和简单定制。create.py:交互式生成,支持循环输入。models/:模型结构定义代码。transformer/、text_encoder/、vae/、clip_model/:已经下载好的本地模型权重。
如果你追求极致画质且显存充裕(24 GB 以上),可以在脚本里把数据类型改成 float32,不过会明显增加显存占用。默认的 bfloat16 是经过权衡的选择,速度和精度都比较均衡。
注意显存和常见问题
显存门槛:推理需要 14-15 GB,GPU 至少 16 GB 才跑得起来,否则就会遇到 CUDA out of memory。使用云 GPU 的话,选 A10、V100、3090/4090 这类实例比较稳妥。尽量不要同时跑其他占用显存的东西。
改了 prompt 没反应:先确认文件保存了,路径没错,有时框架缓存会让人怀疑人生,重启容器是个简单粗暴但有效的方法。
画质还想再高一点:在 <style> 里加上 ultra_detailed、best_quality、8k 这类关键词;如果能设置输出分辨率,也尽量往高调。不过个人经验是,单人加简单背景的表现一般比复杂多角色场景更稳定。

