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

Tweaks.

Browse Source
This commit is contained in:
laurent
2023-09-19 09:08:32 +01:00
parent 49a4fa44bb
commit 576bf7c21f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/wuerstchen/main.rs
View File

@ -252,7 +252,7 @@ fn run(args: Args) -> Result<()> {
&device, &device,
)? )?
}; };
println!("{prior_text_embeddings}"); println!("generated prior text embeddings {prior_text_embeddings:?}");
let text_embeddings = { let text_embeddings = {
let tokenizer = ModelFile::Tokenizer.get(tokenizer)?; let tokenizer = ModelFile::Tokenizer.get(tokenizer)?;
@ -266,7 +266,7 @@ fn run(args: Args) -> Result<()> {
&device, &device,
)? )?
}; };
println!("{prior_text_embeddings}"); println!("generated text embeddings {text_embeddings:?}");
println!("Building the prior."); println!("Building the prior.");
// https://huggingface.co/warp-ai/wuerstchen-prior/blob/main/prior/config.json // https://huggingface.co/warp-ai/wuerstchen-prior/blob/main/prior/config.json