mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00

* correct optional SE layer dimensions. * head_dim instead of num_heads is 32. * update test example output.
* correct optional SE layer dimensions. * head_dim instead of num_heads is 32. * update test example output.