Fixed a type annotation.

#175
Files changed (1) hide show
  1. inference/model.py +1 -1
inference/model.py CHANGED
@@ -62,7 +62,7 @@ class ModelArgs:
62
  o_groups: int = 8
63
  o_lora_rank: int = 1024
64
  window_size: int = 128
65
- compress_ratios: Tuple[int] = (0, 0, 4, 128, 4, 128, 4, 0)
66
  # yarn
67
  compress_rope_theta: float = 40000.0
68
  original_seq_len: int = 0
 
62
  o_groups: int = 8
63
  o_lora_rank: int = 1024
64
  window_size: int = 128
65
+ compress_ratios: Tuple[int, int, int, int, int, int, int, int] = (0, 0, 4, 128, 4, 128, 4, 0)
66
  # yarn
67
  compress_rope_theta: float = 40000.0
68
  original_seq_len: int = 0