🚀 Flux.1-Dev-LoRA-HDR-Realism [實驗版本]
本模型是用於圖像生成的模型,結合了LoRA和HDR技術,能生成具有高動態範圍和逼真效果的圖像。不過當前仍處於訓練階段,效果可能存在不足。
🚀 快速開始
環境設置
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/Flux.1-Dev-LoRA-HDR-Realism"
trigger_word = "HDR"
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
觸發詞使用
你應該使用 HDR
來觸發圖像生成。
模型下載
該模型的權重以Safetensors格式提供。
點擊下載,可在“Files & versions”標籤中獲取。
✨ 主要特性
- 基於
black-forest-labs/FLUX.1-dev
基礎模型,通過LoRA技術進行微調。
- 支持HDR圖像生成,能呈現出高動態範圍和逼真的效果。
- 提供了多個示例圖像,展示了不同場景下的生成效果。
📦 安裝指南
暫未提供具體安裝步驟,可參考上述快速開始部分的代碼示例進行環境設置。
💻 使用示例
基礎用法
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/Flux.1-Dev-LoRA-HDR-Realism"
trigger_word = "HDR"
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
高級用法
在上述基礎用法的基礎上,你可以根據需要調整圖像生成的參數,例如:
pipe.set_parameter("param_name", "param_value")
prompt = f"{trigger_word}, your specific description"
image = pipe(prompt).images[0]
image.save("output.png")
📚 詳細文檔
模型描述
prithivMLmods/Flux.1-Dev-LoRA-HDR-Realism
圖像生成參數如下:
參數 |
值 |
參數 |
值 |
LR Scheduler |
constant |
Noise Offset |
0.03 |
Optimizer |
AdamW |
Multires Noise Discount |
0.1 |
Network Dim |
64 |
Multires Noise Iterations |
10 |
Network Alpha |
32 |
Repeat & Steps |
19 & 2000 |
Epoch |
17 |
Save Every N Epochs |
1 |
標註信息:florence2-en(自然語言 & 英語)
訓練使用的總圖像數:14
最佳尺寸
📄 許可證
本模型使用 creativeml-openrail-m
許可證。
⚠️ 重要提示
該模型仍處於訓練階段,這不是最終版本,可能會包含偽影,並且在某些情況下表現不佳。
💡 使用建議
你可以訪問模型的託管頁面🧨:https://huggingface.co/spaces/prithivMLmods/FLUX-LoRA-DLC ,在該頁面上進行圖像生成的測試和體驗。