Upload folder using huggingface_hub
Browse files
convert_original_to_comfy.py
CHANGED
|
@@ -9,13 +9,15 @@ Requirements:
|
|
| 9 |
pip install torch safetensors
|
| 10 |
|
| 11 |
Usage:
|
|
|
|
| 12 |
python convert_causal_forcing_standalone.py \
|
| 13 |
--input checkpoints/framewise/causal_forcing.pt \
|
| 14 |
--output models/causal_forcing_framewise.safetensors
|
| 15 |
|
|
|
|
| 16 |
python convert_causal_forcing_standalone.py \
|
| 17 |
-
--input checkpoints/
|
| 18 |
-
--output models/
|
| 19 |
--no-ema
|
| 20 |
"""
|
| 21 |
|
|
|
|
| 9 |
pip install torch safetensors
|
| 10 |
|
| 11 |
Usage:
|
| 12 |
+
# Framewise model (uses EMA weights by default):
|
| 13 |
python convert_causal_forcing_standalone.py \
|
| 14 |
--input checkpoints/framewise/causal_forcing.pt \
|
| 15 |
--output models/causal_forcing_framewise.safetensors
|
| 16 |
|
| 17 |
+
# Chunkwise model (uses non-EMA weights):
|
| 18 |
python convert_causal_forcing_standalone.py \
|
| 19 |
+
--input checkpoints/chunkwise/causal_forcing.pt \
|
| 20 |
+
--output models/causal_forcing_chunkwise.safetensors \
|
| 21 |
--no-ema
|
| 22 |
"""
|
| 23 |
|