This commit is contained in:
laurent
2023-09-19 09:08:32 +01:00
parent 49a4fa44bb
commit 576bf7c21f

View File

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