Commit8be0710

committed

add test showing vllm model support check

1 parent9360ef7 commit8be0710Copy full SHA for 8be0710

File tree

-0

lines changed

-0

lines changed

Lines changed: 8 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -263,6 +263,7 @@ mod tests {`
`263`	`263`	`usesuper::*;`
`264`	`264`
`265`	`265`	`#[pg_test]`
	`266`	`+#[ignore ="requires model download"]`
`266`	`267`	`fnvllm_quickstart(){`
`267`	`268`	`crate::bindings::python::activate().unwrap();`
`268`	`269`
`@@ -283,4 +284,11 @@ mod tests {`
`283`	`284`	`let outputs = llm.generate(&prompts,Some(&sampling_params)).unwrap();`
`284`	`285`	`assert_eq!(prompts.len(), outputs.len());`
`285`	`286`	`}`
	`287`	`+`
	`288`	`+#[pg_test]`
	`289`	`+fnmodel_support(){`
	`290`	`+ifletErr(e) =LLMBuilder::new("intfloat/e5-small").build(){`
	`291`	`+assert!(e.to_string().contains("not supported"));`
	`292`	`+}`
	`293`	`+}`
`286`	`294`	`}`

Comments

(0)