File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
test/modelDependent/llama3.1 Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -24,18 +24,18 @@ describe("llama 3.1", () => {
2424 for ( const token of model . iterateAllTokens ( ) ) {
2525 const text = model . detokenize ( [ token ] ) ;
2626
27- if ( text . toLowerCase ( ) . includes ( "hello " ) )
27+ if ( text . toLowerCase ( ) . includes ( "well " ) )
2828 customBias . set ( token , - 0.99 ) ;
2929 else if ( text . toLowerCase ( ) . includes ( "hi" ) )
3030 customBias . set ( token , "never" ) ;
3131 }
3232
33- const res = await chatSession . prompt ( 'Greet me by saying "hello " to me' , {
33+ const res = await chatSession . prompt ( 'Greet me by saying "hi " to me' , {
3434 tokenBias : customBias ,
3535 maxTokens : 100
3636 } ) ;
3737
38- expect ( res . toLowerCase ( ) ) . to . not . include ( "hello " ) ;
38+ expect ( res . toLowerCase ( ) ) . to . not . include ( "well " ) ;
3939 expect ( res . toLowerCase ( ) ) . to . not . include ( "hi " ) ;
4040 expect ( res . toLowerCase ( ) ) . to . not . include ( "hi." ) ;
4141 } ) ;
You can’t perform that action at this time.
0 commit comments