🚀 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许可证。
⚠️ 重要提示
该模型仍处于训练阶段,这不是最终版本,可能会包含瑕疵,并且在某些情况下表现不佳。