🚀 pytorch_lora_weights.safetensors
このモデルは、テキストから画像を生成するためのStable Diffusionアーキテクチャをベースに、Low-Rank Adaptation (LoRA) 技術を用いて微調整されたものです。高品質な人物顔画像を持つCelebA-HQとFFHQのデータセットを使用して学習されています。
🚀 クイックスタート
このモデルを使用することで、テキスト入力から高品質な人物顔画像を生成できます。以下に使用例を示します。
✨ 主な機能
- Stable Diffusionアーキテクチャをベースに、LoRA技術を用いた微調整により、高品質な人物顔画像を生成できます。
- CelebA-HQとFFHQのデータセットを使用して学習されており、多様な人物顔画像を生成できます。
📦 インストール
このモデルの重みはSafetensors形式で提供されています。以下のリンクからダウンロードできます。
ダウンロード
💻 使用例
基本的な使用法
import torch
from diffusers import StableDiffusionPipeline,UNet2DConditionModel
pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to("cuda")
pipeline.load_lora_weights("phil329/face_lora_sd15", weight_name="pytorch_lora_weights.safetensors")
NEGATIVE_PROMPT = "worst quality, low quality, bad anatomy, watermark, text, blurry, cartoon, unreal"
text = 'A young woman with smile, wearing a purple hat.'
lora_image = pipeline(text,negative_prompt=NEGATIVE_PROMPT).images[0]
display(lora_image)
高度な使用法
このモデルでは、異なるプロンプトやネガティブプロンプトを使用することで、様々な画像を生成できます。以下にいくつかのプロンプト例を示します。
import torch
from diffusers import StableDiffusionPipeline,UNet2DConditionModel
pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to("cuda")
pipeline.load_lora_weights("phil329/face_lora_sd15", weight_name="pytorch_lora_weights.safetensors")
NEGATIVE_PROMPT = "worst quality, low quality, bad anatomy, watermark, text, blurry, cartoon, unreal"
prompts = [
'A middle-aged man,beard ,attractive',
'A girl with long blonde hair',
'An young man with curry hair'
]
for prompt in prompts:
lora_image = pipeline(prompt, negative_prompt=NEGATIVE_PROMPT).images[0]
display(lora_image)
📚 ドキュメント
モデルの詳細
このモデルは、Stable Diffusionアーキテクチャをベースに、LoRA技術を用いて微調整されたものです。以下に学習の詳細を示します。
属性 |
详情 |
モデルタイプ |
Stable Diffusionの微調整モデル |
学習データ |
CelebA-HQ (30,000枚の画像), FFHQ (70,000枚の画像) |
解像度 |
512*512(詳細な顔合成のための微調整) |
結果
以下の4つのプロンプトを使用して画像を生成しました。ネガティブプロンプトはコード例と同じです。すべての結果はランダムに生成されており、選別されたものではありません。
- 'A young woman with smile, wearing a purple hat.'
- 'A middle-aged man,beard ,attractive'
- 'A girl with long blonde hair'
- 'An young man with curry hair'
生成結果が良くない場合は、ネガティブプロンプトを追加するか、異なるプロンプトやシードを試してみてください。

📄 ライセンス
このモデルはMITライセンスの下で提供されています。