Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -5,7 +5,7 @@ license: apache-2.0
|
|
| 5 |
|
| 6 |
This model is one of the open-source Diffusion Templates series models from [DiffSynth-Studio](https://github.com/modelscope/DiffSynth-Studio). It is an Aesthetic alignment model that allows adjusting the degree of aesthetic alignment in generated images by modifying the `scale` parameter.
|
| 7 |
|
| 8 |
-
## Results
|
| 9 |
|
| 10 |
> **Prompt:** A cat is sitting on a stone.
|
| 11 |
|
|
@@ -39,12 +39,13 @@ cd DiffSynth-Studio
|
|
| 39 |
pip install -e .
|
| 40 |
```
|
| 41 |
|
| 42 |
-
* Direct inference (requires
|
| 43 |
|
| 44 |
```python
|
| 45 |
from diffsynth.diffusion.template import TemplatePipeline
|
| 46 |
from diffsynth.pipelines.flux2_image import Flux2ImagePipeline, ModelConfig
|
| 47 |
import torch
|
|
|
|
| 48 |
|
| 49 |
pipe = Flux2ImagePipeline.from_pretrained(
|
| 50 |
torch_dtype=torch.bfloat16,
|
|
@@ -165,7 +166,7 @@ image.save("image_Aesthetic_2.5.jpg")
|
|
| 165 |
|
| 166 |
```
|
| 167 |
|
| 168 |
-
## Training
|
| 169 |
|
| 170 |
After installing DiffSynth-Studio, use the following script to start training. For more information, please refer to the [DiffSynth-Studio Documentation](https://diffsynth-studio-doc.readthedocs.io/zh-cn/latest/).
|
| 171 |
|
|
|
|
| 5 |
|
| 6 |
This model is one of the open-source Diffusion Templates series models from [DiffSynth-Studio](https://github.com/modelscope/DiffSynth-Studio). It is an Aesthetic alignment model that allows adjusting the degree of aesthetic alignment in generated images by modifying the `scale` parameter.
|
| 7 |
|
| 8 |
+
## Results
|
| 9 |
|
| 10 |
> **Prompt:** A cat is sitting on a stone.
|
| 11 |
|
|
|
|
| 39 |
pip install -e .
|
| 40 |
```
|
| 41 |
|
| 42 |
+
* Direct inference (requires 40G GPU memory)
|
| 43 |
|
| 44 |
```python
|
| 45 |
from diffsynth.diffusion.template import TemplatePipeline
|
| 46 |
from diffsynth.pipelines.flux2_image import Flux2ImagePipeline, ModelConfig
|
| 47 |
import torch
|
| 48 |
+
```
|
| 49 |
|
| 50 |
pipe = Flux2ImagePipeline.from_pretrained(
|
| 51 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 166 |
|
| 167 |
```
|
| 168 |
|
| 169 |
+
## Training Script
|
| 170 |
|
| 171 |
After installing DiffSynth-Studio, use the following script to start training. For more information, please refer to the [DiffSynth-Studio Documentation](https://diffsynth-studio-doc.readthedocs.io/zh-cn/latest/).
|
| 172 |
|