ART-3D commited on
Commit
880202e
·
verified ·
1 Parent(s): be78101

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +34 -0
README.md ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - 3d-editing
5
+ - trellis
6
+ - controlnet
7
+ ---
8
+
9
+ # PartFlow — model weights
10
+
11
+ Trained ControlNet denoisers for **PartFlow**, a two-stage image-conditioned
12
+ 3D editing model built on [TRELLIS](https://github.com/microsoft/TRELLIS).
13
+
14
+ Inference code: see the PartFlow GitHub repository.
15
+
16
+ ## Contents
17
+
18
+ | Path | Model | Description |
19
+ |---|---|---|
20
+ | `stage1_ss/denoiser_ema0.9999_step0080000.pt` | `ControlledSparseStructureFlowModel` | Stage 1 — sparse-structure ControlNet (EMA, step 80k) |
21
+ | `stage1_ss/config.json` | — | Stage 1 model config |
22
+ | `stage2_slat/denoiser_ema0.9999_step0080000.pt` | `ControlledSLatFlowModel` | Stage 2 — structured-latent ControlNet (EMA, step 80k) |
23
+ | `stage2_slat/config.json` | — | Stage 2 model config |
24
+
25
+ These are EMA weights (decay 0.9999). The frozen DINOv2 encoder and TRELLIS
26
+ SS/SLat decoders are not included — PartFlow fetches them from their official
27
+ repos at inference time.
28
+
29
+ ## Usage
30
+
31
+ ```bash
32
+ python download_weights.py # pulls this repo into ./weights/
33
+ python inference.py --input examples/add_ring_band --output_dir outputs
34
+ ```