Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -8,7 +8,7 @@ tags:
|
|
| 8 |
- pytorch
|
| 9 |
- fcdm
|
| 10 |
- semantic-alignment
|
| 11 |
-
library_name:
|
| 12 |
---
|
| 13 |
|
| 14 |
# data-archetype/semdisdiffae
|
|
@@ -67,9 +67,9 @@ encoder or decoder, enabling efficient inference at any resolution.
|
|
| 67 |
## Quick Start
|
| 68 |
|
| 69 |
```python
|
| 70 |
-
from
|
| 71 |
|
| 72 |
-
model =
|
| 73 |
|
| 74 |
# Encode (returns posterior mode by default)
|
| 75 |
latents = model.encode(images) # [B,3,H,W] in [-1,1] -> [B,128,H/16,W/16]
|
|
@@ -78,7 +78,7 @@ latents = model.encode(images) # [B,3,H,W] in [-1,1] -> [B,128,H/16,W/16]
|
|
| 78 |
recon = model.decode(latents, height=H, width=W)
|
| 79 |
|
| 80 |
# Decode — perceptual sharpening (10 steps + PDG)
|
| 81 |
-
cfg =
|
| 82 |
recon = model.decode(latents, height=H, width=W, inference_config=cfg)
|
| 83 |
|
| 84 |
# Full posterior access
|
|
|
|
| 8 |
- pytorch
|
| 9 |
- fcdm
|
| 10 |
- semantic-alignment
|
| 11 |
+
library_name: fcdm_diffae
|
| 12 |
---
|
| 13 |
|
| 14 |
# data-archetype/semdisdiffae
|
|
|
|
| 67 |
## Quick Start
|
| 68 |
|
| 69 |
```python
|
| 70 |
+
from fcdm_diffae import FCDMDiffAE, FCDMDiffAEInferenceConfig
|
| 71 |
|
| 72 |
+
model = FCDMDiffAE.from_pretrained("data-archetype/semdisdiffae", device="cuda")
|
| 73 |
|
| 74 |
# Encode (returns posterior mode by default)
|
| 75 |
latents = model.encode(images) # [B,3,H,W] in [-1,1] -> [B,128,H/16,W/16]
|
|
|
|
| 78 |
recon = model.decode(latents, height=H, width=W)
|
| 79 |
|
| 80 |
# Decode — perceptual sharpening (10 steps + PDG)
|
| 81 |
+
cfg = FCDMDiffAEInferenceConfig(num_steps=10, pdg=True, pdg_strength=2.0)
|
| 82 |
recon = model.decode(latents, height=H, width=W, inference_config=cfg)
|
| 83 |
|
| 84 |
# Full posterior access
|