AIbase

P

Plushies Pt

由 camenduru 开发
这是一个基于TPU训练并转换为PyTorch的稳定扩散模型，专注于文本生成图像任务，特别适用于毛绒玩具类图像的生成。
文本生成图像
英语
开源协议:Openrail#TPU训练#JAX优化#稳定扩散
下载量 18
发布时间 : 12/24/2022
模型介绍
内容详情
替代品

模型简介

该模型使用JAX和Flax框架在谷歌云TPU上训练，能够根据文本描述生成高质量的毛绒玩具图像。它是首个使用TPU训练并转换为PyTorch的稳定扩散模型。

模型特点

TPU训练
使用谷歌云TPU进行训练，提高了训练效率。
JAX/Flax框架
基于JAX和Flax框架开发，支持高效计算。
稳定扩散架构
采用稳定扩散模型架构，能够生成高质量的图像。
PyTorch兼容
训练后转换为PyTorch格式，便于在更广泛的平台上使用。

模型能力

文本生成图像
毛绒玩具图像生成

使用案例

创意设计
毛绒玩具设计
根据文本描述生成毛绒玩具的设计图像
可生成高质量的毛绒玩具图像样本
电子商务
产品展示图生成
为电商平台自动生成毛绒玩具产品展示图