mehdiPrunaAI commited on
Commit
0439d7a
·
verified ·
1 Parent(s): 31e1652

Update tests/test_distilled_decode.py

Browse files
Files changed (1) hide show
  1. tests/test_distilled_decode.py +2 -6
tests/test_distilled_decode.py CHANGED
@@ -46,16 +46,12 @@ LTX23_VAE = "diffusers/LTX-2.3-Diffusers" # stock decoder (baseline)
46
  PRUNED_VAE = REPO_ROOT # this repo's vae/ folder
47
 
48
  # ~720p for 5 s @ 24 fps. Height/width must be multiples of 64 (2-stage).
49
- HEIGHT, WIDTH, NUM_FRAMES, FPS = 704, 1280, 121, 24.0
50
  SEED = 42
51
  DECODE_WARMUP, DECODE_RUNS = 1, 3 # timing: 1 warm-up + median of 3 runs
52
 
53
  PROMPT = (
54
- "In the video, a man is seen riding a skateboard in an indoor skate park. "
55
- "He is wearing a black cap, a grey t-shirt, and black pants. The skate park "
56
- "has a wooden floor and a ramp. The man performs a trick by jumping and "
57
- "flipping the skateboard under his feet while in the air. He lands on the "
58
- "skateboard and continues to ride it."
59
  )
60
 
61
  DEVICE = "cuda"
 
46
  PRUNED_VAE = REPO_ROOT # this repo's vae/ folder
47
 
48
  # ~720p for 5 s @ 24 fps. Height/width must be multiples of 64 (2-stage).
49
+ HEIGHT, WIDTH, NUM_FRAMES, FPS = 1088, 1920, 121, 24.0
50
  SEED = 42
51
  DECODE_WARMUP, DECODE_RUNS = 1, 3 # timing: 1 warm-up + median of 3 runs
52
 
53
  PROMPT = (
54
+ "The video shows a hockey player in a green jersey and blue helmet skating on the ice with a hockey stick. The player is seen moving around the ice, passing the puck to another player who is also wearing a green jersey and blue helmet. The player in the green jersey is seen skating away from the camera, and then turning around to face the camera. The ice rink is surrounded by boards with advertisements, and there are other players in the background. The player in the green jersey is wearing black gloves and black skates. The player in the green jersey is also seen skating towards the camera and then away from the camera again."
 
 
 
 
55
  )
56
 
57
  DEVICE = "cuda"