Upload folder using huggingface_hub
Browse files- README.md +2 -4
- README_from_modelscope.md +4 -0
README.md
CHANGED
|
@@ -3,7 +3,7 @@ license: apache-2.0
|
|
| 3 |
---
|
| 4 |
# Templates - Brightness Adjustment (FLUX.2-klein-base-4B)
|
| 5 |
|
| 6 |
-
This model is part of the open-source Diffusion Templates series
|
| 7 |
|
| 8 |
## Results
|
| 9 |
|
|
@@ -39,13 +39,12 @@ cd DiffSynth-Studio
|
|
| 39 |
pip install -e .
|
| 40 |
```
|
| 41 |
|
| 42 |
-
* Direct inference
|
| 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,
|
|
@@ -94,7 +93,6 @@ image.save("image_Brightness_dark.jpg")
|
|
| 94 |
from diffsynth.diffusion.template import TemplatePipeline
|
| 95 |
from diffsynth.pipelines.flux2_image import Flux2ImagePipeline, ModelConfig
|
| 96 |
import torch
|
| 97 |
-
```
|
| 98 |
|
| 99 |
```python
|
| 100 |
vram_config = {
|
|
|
|
| 3 |
---
|
| 4 |
# Templates - Brightness Adjustment (FLUX.2-klein-base-4B)
|
| 5 |
|
| 6 |
+
This model is part of the open-source Diffusion Templates series by [DiffSynth-Studio](https://github.com/modelscope/DiffSynth-Studio). It is a brightness adjustment model capable of customizing and modifying the brightness of images.
|
| 7 |
|
| 8 |
## Results
|
| 9 |
|
|
|
|
| 39 |
pip install -e .
|
| 40 |
```
|
| 41 |
|
| 42 |
+
* Direct inference (requires 40GB 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 |
pipe = Flux2ImagePipeline.from_pretrained(
|
| 50 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 93 |
from diffsynth.diffusion.template import TemplatePipeline
|
| 94 |
from diffsynth.pipelines.flux2_image import Flux2ImagePipeline, ModelConfig
|
| 95 |
import torch
|
|
|
|
| 96 |
|
| 97 |
```python
|
| 98 |
vram_config = {
|
README_from_modelscope.md
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
---
|
|
|
|
|
|
|
| 2 |
frameworks:
|
| 3 |
- Pytorch
|
| 4 |
license: Apache License 2.0
|
|
@@ -177,3 +179,5 @@ accelerate launch examples/flux2/model_training/train.py \
|
|
| 177 |
--use_gradient_checkpointing \
|
| 178 |
--find_unused_parameters
|
| 179 |
```
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
base_model:
|
| 3 |
+
- ""
|
| 4 |
frameworks:
|
| 5 |
- Pytorch
|
| 6 |
license: Apache License 2.0
|
|
|
|
| 179 |
--use_gradient_checkpointing \
|
| 180 |
--find_unused_parameters
|
| 181 |
```
|
| 182 |
+
|
| 183 |
+
|