Skip to content

Commit da2142b

Browse files
authored
Fix module forward calls after api changes.
Differential Revision: D61827911 Pull Request resolved: #4922
1 parent 7efdfc0 commit da2142b

File tree

1 file changed

+3
-3
lines changed
  • examples/qualcomm/qaihub_scripts/stable_diffusion/runner

1 file changed

+3
-3
lines changed

examples/qualcomm/qaihub_scripts/stable_diffusion/runner/runner.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -377,11 +377,11 @@ Error Runner::generate(std::string prompt) {
377377
Tensor uncond_emb_tensor = managed_uncond_emb.get_aliasing_tensor();
378378
modules_[0]->set_output_data_ptr(cond_emb_tensor, 0);
379379
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);
381381
stats_.text_encoder_execution_time += (util::time_in_ms() - encoder_start);
382382
modules_[0]->set_output_data_ptr(uncond_emb_tensor, 0);
383383
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);
385385
stats_.text_encoder_execution_time += (util::time_in_ms() - encoder_start);
386386

387387
// Initialize unet parameters
@@ -533,7 +533,7 @@ Error Runner::generate(std::string prompt) {
533533

534534
modules_[2]->set_output_data_ptr(output_tensor, 0);
535535
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);
537537
stats_.vae_execution_time = (util::time_in_ms() - start_vae_execution);
538538
stats_.generate_end_ms = util::time_in_ms();
539539

0 commit comments

Comments
 (0)