File size: 3,257 Bytes
7408d34
 
 
 
 
 
3370c66
 
7408d34
3370c66
 
 
 
 
 
7408d34
bf2a287
3370c66
 
 
 
 
 
 
bf2a287
 
3370c66
 
 
 
 
 
bf2a287
 
7408d34
 
 
 
 
 
 
927f7db
7408d34
927f7db
7408d34
dee7dcb
7408d34
927f7db
7408d34
927f7db
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7408d34
 
 
 
 
3370c66
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
---
tags:
- text-to-image
- lora
- diffusers
- template:diffusion-lora
- 3D
- Portrait
widget:
- text: >-
    A 3D Portrait cartoon depiction of a woman with long dark brown hair and
    brown eyes stands in front of a window. She is wearing a gray turtleneck,
    with her left hand resting on her head. Her right hand is touching her face,
    adding a touch of warmth to the scene. The background is a stark white wall,
    with a window on the left side of the image.
  output:
    url: images/1.webp
- text: >-
    A 3D cartoon-style portrait of a man with short, wavy black hair and blue
    eyes, standing against a soft pastel-colored background. He is wearing a
    casual navy blue hoodie, with his right arm crossed over his chest, and his
    left hand raised to touch his chin thoughtfully. The background includes a
    large, arched window with light streaming through, creating a warm, cozy
    atmosphere
  output:
    url: images/2.webp
- text: >-
    A 3D cartoon-style portrait of a young woman with medium-length blonde hair
    and green eyes, sitting in a cozy cafe. She is wearing a light pink sweater
    and holding a warm cup of coffee close to her face with both hands, creating
    a peaceful vibe. The background shows a wooden table and a window with
    raindrops, giving a cozy, rainy-day feel
  output:
    url: images/3.webp
base_model: black-forest-labs/FLUX.1-dev
instance_prompt: 3D Portrait
license: creativeml-openrail-m
---
# Castor-3D-Portrait-Flux-LoRA

<Gallery />
**The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.**

## Model description 

**prithivMLmods/Castor-3D-Portrait-Flux-LoRA**

Image Processing Parameters 

| Parameter                 | Value  | Parameter                 | Value  |
|---------------------------|--------|---------------------------|--------|
| 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           | 20 & 2.5K |
| Epoch                     | 15     | Save Every N Epochs       | 1      |

    Labeling: florence2-en(natural language & English)
    
    Total Images Used for Training : 19

## Setting Up
```
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/Castor-3D-Portrait-Flux-LoRA"
trigger_word = "3D Portrait"  # Leave trigger_word blank if not used.
pipe.load_lora_weights(lora_repo)

device = torch.device("cuda")
pipe.to(device)
```
## App File Structure

/project-root/

β”œβ”€β”€ .gitattributes         
β”œβ”€β”€ README.md      
β”œβ”€β”€ app.py             
β”œβ”€β”€ pythonproject.py     

## Trigger words 🧨

> [!WARNING]
> **Trigger words:**  You should use `3D Portrait` to trigger the image generation.

## Download model

Weights for this model are available in Safetensors format.

[Download](/prithivMLmods/Castor-3D-Portrait-Flux-LoRA/tree/main) them in the Files & versions tab.