Buckets:

mishig's picture
|
download
raw
2.54 kB
# FreeU
[FreeU](https://huggingface.co/papers/2309.11497) melhora o detalhe da imagem ao reequilibrar a quantidade de recursos que o decodificador do UNet extrai dos recursos do backbone em comparação com os recursos das conexões de atalho. As conexões de atalho podem abafar os recursos semânticos do backbone, o que gera detalhes não naturais na saída. O FreeU não requer treinamento e pode ser ativado ou desativado no momento da inferência para pipelines de texto para imagem e texto para vídeo.
> [!NOTE]
> FreeU funciona apenas com pipelines baseados em UNet, como Stable Diffusion, SDXL e AnimateDiff. Não é suportado por pipelines baseados em transformadores, como Flux ou Qwen-Image.
Use o método [enable_freeu()](/docs/diffusers/v0.43.0/pt/api/pipelines/overview#diffusers.StableDiffusionMixin.enable_freeu) no seu pipeline e configure os fatores de escala. `b1` e `b2` amplificam os recursos do backbone, e `s1` e `s2` atenuam os recursos de skip. Os `1` e `2` referem-se às duas primeiras etapas de upsampling do decodificador da UNet. Consulte o repositório [FreeU](https://github.com/ChenyangSi/FreeU#parameters) para hiperparâmetros de referência para diferentes modelos.
Comece com os valores do repositório para um modelo. Para ajustar para outros modelos, mantenha `s1=0.9` e `s2=0.2` e ajuste primeiro `b1` e `b2`. Definir os quatro fatores como `1.0` é o mesmo que desativar o FreeU. Valores maiores de `b` fortalecem o efeito, mas podem suavizar demais a textura fina, e reduzir `s1` e `s2` contrabalança isso.
```py
import torch
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0", dtype=torch.float16,
).to("cuda") # or "mps", "xpu", "cpu"
pipeline.enable_freeu(s1=0.9, s2=0.2, b1=1.3, b2=1.4)
generator = torch.Generator(device="cpu").manual_seed(13)
prompt = "A squirrel eating a burger"
image = pipeline(prompt, generator=generator).images[0]
image
```
FreeU desativado
FreeU habilitado
Chame o método [disable_freeu()](/docs/diffusers/v0.43.0/pt/api/pipelines/overview#diffusers.StableDiffusionMixin.disable_freeu) para desativar o FreeU.
```py
pipeline.disable_freeu()
```
## Próximas etapas
- Consulte a referência da API [enable_freeu()](/docs/diffusers/v0.43.0/pt/api/pipelines/overview#diffusers.StableDiffusionMixin.enable_freeu) para as descrições completas dos parâmetros.
- Experimente o FreeU em vídeo com [AnimateDiff](../api/pipelines/animatediff).

Xet Storage Details

Size:
2.54 kB
·
Xet hash:
51ad1920e029da72f2cab49ecec97c901fd9ee517ffd77ea50a6a1b9a23451e8

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.