@@ -1185,7 +1185,7 @@ describe.concurrent("HfInference", () => {
11851185 "meta-llama/Llama-3.2-3B-Instruct" : "meta-llama/Llama-3.2-3B-Instruct" ,
11861186 "meta-llama/Llama-3.3-70B-Instruct" : "meta-llama/Llama-3.3-70B-Instruct" ,
11871187 "stabilityai/stable-diffusion-2" : "stabilityai/stable-diffusion-2" ,
1188- "meta-llama/Llama-3.1-405B-BASE-FP8 " : "meta-llama/Llama-3.1-405B-BASE-FP8 " ,
1188+ "meta-llama/Llama-3.1-405B" : "meta-llama/Llama-3.1-405B" ,
11891189 } ;
11901190
11911191 it ( "chatCompletion - hyperbolic" , async ( ) => {
@@ -1227,20 +1227,18 @@ describe.concurrent("HfInference", () => {
12271227 const res = await client . textToImage ( {
12281228 model : "stabilityai/stable-diffusion-2" ,
12291229 provider : "hyperbolic" ,
1230- inputs : " award winning high resolution photo of a giant tortoise",
1230+ messages : [ { role : "user" , content : " award winning high resolution photo of a giant tortoise" } ] ,
12311231 } ) ;
12321232 expect ( res ) . toBeInstanceOf ( Blob ) ;
12331233 } ) ;
12341234
12351235 it ( "textGeneration" , async ( ) => {
12361236 const res = await client . textGeneration ( {
1237- model : "meta-llama/Llama-3.1-405B-BASE-FP8 " ,
1237+ model : "meta-llama/Llama-3.1-405B" ,
12381238 provider : "hyperbolic" ,
1239- inputs : "Paris is" ,
1240- parameters : {
1241- temperature : 0 ,
1242- max_tokens : 10 ,
1243- } ,
1239+ messages : [ { role : "user" , content : "Paris is" } ] ,
1240+ temperature : 0 ,
1241+ max_tokens : 10 ,
12441242 } ) ;
12451243 expect ( res ) . toMatchObject ( { generated_text : " city of love" } ) ;
12461244 } ) ;
0 commit comments