update fmt
This commit is contained in:
+6
-4
@@ -1,7 +1,10 @@
|
||||
use std::{pin::pin, time::Instant};
|
||||
use aha::{
|
||||
chat::ChatCompletionParameters,
|
||||
models::{GenerateModel, lfm2::generate::Lfm2GenerateModel},
|
||||
};
|
||||
use anyhow::Result;
|
||||
use aha::{chat::ChatCompletionParameters, models::{GenerateModel, lfm2::generate::Lfm2GenerateModel}};
|
||||
use rocket::futures::StreamExt;
|
||||
use std::{pin::pin, time::Instant};
|
||||
|
||||
#[test]
|
||||
fn lfm2_generate() -> Result<()> {
|
||||
@@ -44,7 +47,6 @@ fn lfm2_generate() -> Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
|
||||
#[tokio::test]
|
||||
async fn lfm2_stream() -> Result<()> {
|
||||
// test with cuda: RUST_BACKTRACE=1 cargo test -F cuda --test test_lfm2 lfm2_stream -r -- --nocapture
|
||||
@@ -88,4 +90,4 @@ async fn lfm2_stream() -> Result<()> {
|
||||
println!("Time elapsed in generate is: {:?}", i_duration);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user