Alignprop Trl Aesthetics

🚀 AlignProp Aesthetics TRL

這是一個基於Stable Diffusion v1.5微調的文生圖模型，在動物數據集上使用美學獎勵函數，通過AlignProp（https://align-prop.github.io/）的獎勵反向傳播進行訓練。

🚀 快速開始

你可以在下面的示例中輸入關鍵詞，查看模型生成的圖像效果：

✨ 主要特性

• 基於Stable Diffusion v1.5進行微調。
• 在動物數據集上使用美學獎勵函數進行訓練。
• 通過AlignProp的獎勵反向傳播機制優化模型。

📚 詳細文檔

模型描述

這是一個基於Stable Diffusion v1.5微調的文生圖模型。它在動物數據集上使用美學獎勵函數，並通過AlignProp（https://align-prop.github.io/）的獎勵反向傳播進行訓練。該模型使用TRL倉庫中的代碼進行訓練。

觸發詞

你可以使用以下關鍵詞觸發圖像生成：

• alignprop
• reward backpropagation
• rlhf
• diffusion
• reward gradients

下載模型

該模型的權重以Safetensors和PyTorch格式提供。 你可以在Files & versions標籤頁下載模型權重。

📄 許可證

本模型採用Apache 2.0許可證。