/
rustwizard
/
etch
Обзор
Документация
Войти
/
rustwizard
/
etch
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
src/flux/model.rs
27 строк
729 B
Rust Wizard
refactor: extract flux module (model + gguf + run)
27 май 2026, 17:03
Верифицирован
27 май 2026, 17:03
bfd7fb0
Код
Авторство
О чём код?
use candle_core::Tensor; use candle_transformers::models::flux; pub enum FluxModel { Full(Box<flux::model::Flux>), Quantized(Box<flux::quantized_model::Flux>), } impl flux::WithForward for FluxModel { fn forward( &self, img: &Tensor, img_ids: &Tensor, txt: &Tensor, txt_ids: &Tensor, timesteps: &Tensor, y: &Tensor, guidance: Option<&Tensor>, ) -> candle_core::Result<Tensor> { match self { FluxModel::Full(m) => m.forward(img, img_ids, txt, txt_ids, timesteps, y, guidance), FluxModel::Quantized(m) => { m.forward(img, img_ids, txt, txt_ids, timesteps, y, guidance) } } } }