data-archetype commited on
Commit
1747d1e
·
verified ·
1 Parent(s): 433bab6

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -8,7 +8,7 @@ tags:
8
  - pytorch
9
  - fcdm
10
  - semantic-alignment
11
- library_name: capacitor_diffae
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 capacitor_diffae import CapacitorDiffAE, CapacitorDiffAEInferenceConfig
71
 
72
- model = CapacitorDiffAE.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,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 = CapacitorDiffAEInferenceConfig(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
 
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