mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 11:08:52 +00:00
More scaffolding, now need to implement matmul (for precompute_cos_sin to work).
This commit is contained in:
@ -308,7 +308,7 @@ fn main() -> anyhow::Result<()> {
|
||||
| Which::L70b
|
||||
| Which::L70bChat => 8,
|
||||
};
|
||||
ModelWeights::from_ggml(model, args.gqa.unwrap_or(default_gqa))?
|
||||
ModelWeights::from_ggml(model, args.gqa.unwrap_or(default_gqa), &device)?
|
||||
}
|
||||
};
|
||||
println!("model built");
|
||||
|
Reference in New Issue
Block a user