File tree Expand file tree Collapse file tree 1 file changed +26
-0
lines changed
Expand file tree Collapse file tree 1 file changed +26
-0
lines changed Original file line number Diff line number Diff line change 1+ require_relative "helper"
2+
3+ class TestModel < TestBase
4+ def test_model
5+ whisper = Whisper ::Context . new ( MODEL )
6+ assert_instance_of Whisper ::Model , whisper . model
7+ end
8+
9+ def test_attributes
10+ whisper = Whisper ::Context . new ( MODEL )
11+ model = whisper . model
12+
13+ assert_equal 51864 , model . n_vocab
14+ assert_equal 1500 , model . n_audio_ctx
15+ assert_equal 512 , model . n_audio_state
16+ assert_equal 8 , model . n_audio_head
17+ assert_equal 6 , model . n_audio_layer
18+ assert_equal 448 , model . n_text_ctx
19+ assert_equal 512 , model . n_text_state
20+ assert_equal 8 , model . n_text_head
21+ assert_equal 6 , model . n_text_layer
22+ assert_equal 80 , model . n_mels
23+ assert_equal 1 , model . ftype
24+ assert_equal "base" , model . type
25+ end
26+ end
You can’t perform that action at this time.
0 commit comments