Move the llama2-c model in transformers. (#1205)

This commit is contained in:
Laurent Mazare
2023-10-28 17:51:19 +02:00
committed by GitHub
parent 612f5b8156
commit 95a857cf57
6 changed files with 12 additions and 9 deletions

View File

@ -6,10 +6,10 @@ extern crate accelerate_src;
#[cfg(feature = "mkl")]
extern crate intel_mkl_src;
mod model;
mod qmodel;
use candle_transformers::models::llama2_c as model;
use candle_transformers::models::llama2_c_weights as weights;
use candle_transformers::models::quantized_llama2_c as qmodel;
mod training;
mod weights;
use clap::{Parser, Subcommand};
use anyhow::{Error as E, Result};