🚀 F-Meme.FLUX.1-Dev
F-Meme.FLUX.1-Dev是一個文本到圖像的模型,可用於生成各種有趣的表情包圖像。該模型基於特定的基礎模型進行訓練,目前仍處於訓練階段,未來會有更好的表現。
🚀 快速開始
環境設置
import torch
from pipelines import DiffusionPipeline
base_model = "black-forest-labs/FLUX.1-dev"
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
lora_repo = "prithivMLmods/F-Meme"
trigger_word = "meme"
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
觸發詞使用
你應該使用 meme
來觸發圖像生成。
模型下載
此模型的權重以Safetensors格式提供。
點擊下載,可在“文件與版本”選項卡中獲取。
✨ 主要特性
- 文本到圖像轉換:能夠根據輸入的文本描述生成相應的表情包圖像。
- LoRA技術:基於LoRA技術進行訓練,提高模型的訓練效率和靈活性。
📦 安裝指南
安裝步驟已在“快速開始”部分的環境設置中給出,請參考相應的代碼示例。
💻 使用示例
基礎用法
import torch
from pipelines import DiffusionPipeline
base_model = "black-forest-labs/FLUX.1-dev"
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
lora_repo = "prithivMLmods/F-Meme"
trigger_word = "meme"
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
text_description = "meme, A medium-sized painting of a white T-rex in the middle of a dark, stormy night. The t-rex is facing towards the left side of the frame, its head turned towards the right. Its mouth is open, revealing its sharp teeth. A rooster is standing in the foreground of the painting, with a red cap on its head. The roosters head is turned to the right, and the word 'Remember who you are' is written in white text above it. The background is a deep blue, with dark gray clouds and a crescent moon in the upper left corner of the image. There are mountains in the background, and a few other animals can be seen in the lower right corner."
image = pipe(text_description).images[0]
image.show()
📚 詳細文檔
模型描述
prithivMLmods/F-Meme
圖像處理參數
參數 |
詳情 |
LR調度器 |
constant |
優化器 |
AdamW |
網絡維度 |
64 |
網絡Alpha |
32 |
訓練輪數 |
10 |
噪聲偏移 |
0.03 |
多分辨率噪聲折扣 |
0.1 |
多分辨率噪聲迭代次數 |
10 |
重複與步數 |
20 & 2200 |
每N輪保存一次 |
1 |
標註:florence2-en(自然語言 & 英語)
用於訓練的總圖像數:10
最佳尺寸
- 768 x 1024(最佳)
- 1024 x 1024(默認)
📄 許可證
本模型採用creativeml-openrail-m許可證。
⚠️ 重要提示
該模型仍處於訓練階段,這不是最終版本,可能會包含瑕疵,並且在某些情況下表現不佳。