dn6 HF Staff commited on
Commit
f0dc16f
·
verified ·
1 Parent(s): 5378204

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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
  ```