huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 04:10:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Module Docs (#2620)

Browse Source 
* update bert docs

* update based

* update bigcode

* add pixtral

* add flux as well
This commit is contained in:
zachcp
2024-11-16 03:09:17 -05:00
committed by GitHub
parent 00d8a0c178
commit a3f200e369
5 changed files with 126 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
candle-transformers/src/models/flux/mod.rs
View File

@ -1,10 +1,26 @@
//! Flux Model
//!
//! Flux is a series of text-to-image generation models based on diffusion transformers.
//! Flux is a 12B rectified flow transformer capable of generating images from text descriptions.
//!
//! - [GH Link](https://github.com/black-forest-labs/flux)
//! - Transformers Python [reference implementation](https://github.com/huggingface/transformers/blob/5af7d41e49bbfc8319f462eb45253dcb3863dfb7/src/transformers/models/chinese_clip/modeling_chinese_clip.py)
//! - [Hugging Face Model](https://huggingface.co/black-forest-labs/FLUX.1-schnell)
//! - [GitHub Repository](https://github.com/black-forest-labs/flux)
//! - [Blog Post](https://blackforestlabs.ai/announcing-black-forest-labs/)
//!
//! # Usage
//!
//! ```bash
//! cargo run --features cuda \
//! --example flux -r -- \
//! --height 1024 --width 1024 \
//! --prompt "a rusty robot walking on a beach holding a small torch, \
//! the robot has the word \"rust\" written on it, high quality, 4k"
//! ```
//!
//! <div align=center>
//! <img src="https://github.com/huggingface/candle/raw/main/candle-examples/examples/flux/assets/flux-robot.jpg" alt="" width=320>
//! </div>
//!
use candle::{Result, Tensor};
pub trait WithForward {