Text-to-Image
Diffusers
lora
template:diffusion-lora
prithivMLmods commited on
Commit
ebc161d
·
verified ·
1 Parent(s): a658b24

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -1
README.md CHANGED
@@ -36,12 +36,53 @@ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICE
36
 
37
  <Gallery />
38
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
 
40
  ## Trigger words
41
 
42
  You should use `3d station toon` to trigger the image generation.
43
 
44
-
45
  ## Download model
46
 
47
  Weights for this model are available in Safetensors format.
 
36
 
37
  <Gallery />
38
 
39
+ # Model description for 3d-Station-Toon
40
+
41
+ Image Processing Parameters
42
+
43
+ | Parameter | Value | Parameter | Value |
44
+ |---------------------------|--------|---------------------------|--------|
45
+ | LR Scheduler | constant | Noise Offset | 0.03 |
46
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
47
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
48
+ | Network Alpha | 32 | Repeat & Steps | 25 & 3790 |
49
+ | Epoch | 29 | Save Every N Epochs | 1 |
50
+
51
+ Labeling: florence2-en(natural language & English)
52
+
53
+ Total Images Used for Training : 36
54
+
55
+ ## Best Dimensions & Inference
56
+
57
+ | **Dimensions** | **Aspect Ratio** | **Recommendation** |
58
+ |-----------------|------------------|---------------------------|
59
+ | 1280 x 832 | 3:2 | Best |
60
+ | 1024 x 1024 | 1:1 | Default |
61
+
62
+ ### Inference Range
63
+
64
+ - **Recommended Inference Steps:** 30–35
65
+
66
+ ## Setting Up
67
+ ```python
68
+ import torch
69
+ from pipelines import DiffusionPipeline
70
+
71
+ base_model = "black-forest-labs/FLUX.1-dev"
72
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
73
+
74
+ lora_repo = "strangerzonehf/3d-Station-Toon"
75
+ trigger_word = "3d station toon"
76
+ pipe.load_lora_weights(lora_repo)
77
+
78
+ device = torch.device("cuda")
79
+ pipe.to(device)
80
+ ```
81
 
82
  ## Trigger words
83
 
84
  You should use `3d station toon` to trigger the image generation.
85
 
 
86
  ## Download model
87
 
88
  Weights for this model are available in Safetensors format.