File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
examples/qualcomm/qaihub_scripts/stable_diffusion/runner Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -377,11 +377,11 @@ Error Runner::generate(std::string prompt) {
377
377
Tensor uncond_emb_tensor = managed_uncond_emb.get_aliasing_tensor ();
378
378
modules_[0 ]->set_output_data_ptr (cond_emb_tensor, 0 );
379
379
long encoder_start = util::time_in_ms ();
380
- auto cond_res = modules_[0 ]->forward ({ cond_tokens_tensor} );
380
+ auto cond_res = modules_[0 ]->forward (cond_tokens_tensor);
381
381
stats_.text_encoder_execution_time += (util::time_in_ms () - encoder_start);
382
382
modules_[0 ]->set_output_data_ptr (uncond_emb_tensor, 0 );
383
383
encoder_start = util::time_in_ms ();
384
- auto uncond_res = modules_[0 ]->forward ({ uncond_tokens_tensor} );
384
+ auto uncond_res = modules_[0 ]->forward (uncond_tokens_tensor);
385
385
stats_.text_encoder_execution_time += (util::time_in_ms () - encoder_start);
386
386
387
387
// Initialize unet parameters
@@ -533,7 +533,7 @@ Error Runner::generate(std::string prompt) {
533
533
534
534
modules_[2 ]->set_output_data_ptr (output_tensor, 0 );
535
535
long start_vae_execution = util::time_in_ms ();
536
- auto vae_res = modules_[2 ]->forward ({ vae_input_tensor} );
536
+ auto vae_res = modules_[2 ]->forward (vae_input_tensor);
537
537
stats_.vae_execution_time = (util::time_in_ms () - start_vae_execution);
538
538
stats_.generate_end_ms = util::time_in_ms ();
539
539
You can’t perform that action at this time.
0 commit comments