Plushies Pt
這是一個基於TPU訓練並轉換為PyTorch的穩定擴散模型,專注於文本生成圖像任務,特別適用於毛絨玩具類圖像的生成。
下載量 18
發布時間 : 12/24/2022
模型概述
該模型使用JAX和Flax框架在谷歌雲TPU上訓練,能夠根據文本描述生成高質量的毛絨玩具圖像。它是首個使用TPU訓練並轉換為PyTorch的穩定擴散模型。
模型特點
TPU訓練
使用谷歌雲TPU進行訓練,提高了訓練效率。
JAX/Flax框架
基於JAX和Flax框架開發,支持高效計算。
穩定擴散架構
採用穩定擴散模型架構,能夠生成高質量的圖像。
PyTorch兼容
訓練後轉換為PyTorch格式,便於在更廣泛的平臺上使用。
模型能力
文本生成圖像
毛絨玩具圖像生成
使用案例
創意設計
毛絨玩具設計
根據文本描述生成毛絨玩具的設計圖像
可生成高質量的毛絨玩具圖像樣本
電子商務
產品展示圖生成
為電商平臺自動生成毛絨玩具產品展示圖
精選推薦AI模型
Llama 3 Typhoon V1.5x 8b Instruct
專為泰語設計的80億參數指令模型,性能媲美GPT-3.5-turbo,優化了應用場景、檢索增強生成、受限生成和推理任務
大型語言模型
Transformers 支持多種語言

L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一個基於SODA數據集訓練的超小型對話模型,專為邊緣設備推理設計,體積僅為Cosmo-3B模型的2%左右。
對話系統
Transformers 英語

C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基於RoBERTa架構的中文抽取式問答模型,適用於從給定文本中提取答案的任務。
問答系統 中文
R
uer
2,694
98