Oneflow-Inc / libai

LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training
https://libai.readthedocs.io
Apache License 2.0
391 stars 55 forks source link

Gpt2 inference #426

Closed xiezipeng-ML closed 1 year ago

xiezipeng-ML commented 1 year ago

这里是这个demo出来的图片: prompt:a dog, baroque painting, beautiful detailed intricate insanely detailed octane render trending on artstation, 8 k artistic photography, photorealistic, soft natural volumetric cinematic perfect light, chiaroscuro, award - winning photograph 03097c46718e66a57a8d0ccf6a8fb30

xiezipeng-ML commented 1 year ago

oneflow版的权重训练完成:

使用训练6.38万个样本的oneflow权重推理测试:

test1

test2

@CPFLAME