yue-maincode commited on
Commit
775627e
·
verified ·
1 Parent(s): b270811

Model card: require transformers >= 4.56

Browse files

The remote code passes `dtype=` to the inner `from_pretrained`. That keyword is only read from transformers 4.56.0 onward; 4.51-4.55 read only `torch_dtype`, so the requested bfloat16 was silently dropped and `upper_block` loaded in fp32 at roughly double the memory, with no warning.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -89,7 +89,7 @@ print(tok.decode(out[0], skip_special_tokens=True))
89
  The remote code loads the lower block from `Qwen/Qwen3-8B-Base`, the upper
90
  block from `upper_block/` and the bridge from `bridge.safetensors`, truncates both parents to the retained layers and
91
  runs them through their standard `transformers` forward passes with one KV cache per block. `generate` supports greedy
92
- decoding and nucleus sampling (`do_sample`, `temperature`, `top_p`). Requires `transformers >= 4.51` (Qwen3 support).
93
 
94
  ## Limitations
95
 
 
89
  The remote code loads the lower block from `Qwen/Qwen3-8B-Base`, the upper
90
  block from `upper_block/` and the bridge from `bridge.safetensors`, truncates both parents to the retained layers and
91
  runs them through their standard `transformers` forward passes with one KV cache per block. `generate` supports greedy
92
+ decoding and nucleus sampling (`do_sample`, `temperature`, `top_p`). Requires `transformers >= 4.56` (Qwen3 support).
93
 
94
  ## Limitations
95