huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move image tensor to device in trocr example (#2063)

Browse Source 
Signed-off-by: Harry Stern <harry@harrystern.net>
This commit is contained in:
Harry Stern
2024-04-15 00:50:32 -04:00
committed by GitHub
parent c449f65b12
commit c119600d6e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/trocr/main.rs
View File

@ -115,7 +115,7 @@ pub fn main() -> anyhow::Result<()> {
let processor = image_processor::ViTImageProcessor::new(&processor_config); let processor = image_processor::ViTImageProcessor::new(&processor_config);
let image = vec![args.image.as_str()]; let image = vec![args.image.as_str()];
let image = processor.preprocess(image)?; let image = processor.preprocess(image)?.to_device(&device)?;
let encoder_xs = model.encoder().forward(&image)?; let encoder_xs = model.encoder().forward(&image)?;