TalmajM commited on
Commit
087bd5e
·
verified ·
1 Parent(s): 60005bd

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. convert_original_to_comfy.py +4 -2
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/framewise/causal_forcing.pt \
18
- --output models/causal_forcing_framewise.safetensors \
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