Modeld: cannot run without wide img

This commit is contained in:
Bruce Wayne 2024-09-30 00:05:28 -07:00
parent 19bd96004e
commit d4e257213d
1 changed files with 1 additions and 2 deletions

View File

@ -96,8 +96,7 @@ class ModelState:
# if getCLBuffer is not None, frame will be None
self.model.setInputBuffer("input_imgs", self.frame.prepare(buf, transform.flatten(), self.model.getCLBuffer("input_imgs")))
if wbuf is not None:
self.model.setInputBuffer("big_input_imgs", self.wide_frame.prepare(wbuf, transform_wide.flatten(), self.model.getCLBuffer("big_input_imgs")))
self.model.setInputBuffer("big_input_imgs", self.wide_frame.prepare(wbuf, transform_wide.flatten(), self.model.getCLBuffer("big_input_imgs")))
if prepare_only:
return None