Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -57,10 +57,14 @@ recon_image.save("reconstituted.png")
|
|
| 57 |
|
| 58 |
```py
|
| 59 |
import torch
|
| 60 |
-
from diffusers import AutoModel
|
| 61 |
|
| 62 |
device = torch.device("cuda")
|
| 63 |
tiny_vae = AutoModel.from_pretrained(
|
| 64 |
"fal/FLUX.2-Tiny-AutoEncoder", trust_remote_code=True, torch_dtype=torch.bfloat16
|
| 65 |
).to(device)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
```
|
|
|
|
| 57 |
|
| 58 |
```py
|
| 59 |
import torch
|
| 60 |
+
from diffusers import AutoModel, Flux2Pipeline
|
| 61 |
|
| 62 |
device = torch.device("cuda")
|
| 63 |
tiny_vae = AutoModel.from_pretrained(
|
| 64 |
"fal/FLUX.2-Tiny-AutoEncoder", trust_remote_code=True, torch_dtype=torch.bfloat16
|
| 65 |
).to(device)
|
| 66 |
+
|
| 67 |
+
pipe = Flux2Pipeline.from_pretrained(
|
| 68 |
+
"black-forest-labs/FLUX.2-dev", vae=tiny_vae, torch_dtype=torch.bfloat16
|
| 69 |
+
).to(device)
|
| 70 |
```
|