This commit is contained in:
royjr
2026-08-22 16:50:54 -04:00
parent 13c8234c96
commit a697daa669
2 changed files with 2 additions and 5 deletions
+1 -3
View File
@@ -146,9 +146,7 @@ class ModelState(ModelStateBase):
ModelStateBase.__init__(self)
input_devices = get_tg_input_devices(PROCESS_NAME, usbgpu)
self.WARP_DEV, self.QUEUE_DEV = input_devices['WARP_DEV'], input_devices['QUEUE_DEV']
pkl_path = modeld_pkl_path(usbgpu)
stream = open_with_progress(pkl_path) if usbgpu else open_file_chunked(pkl_path)
jits = load_oob(stream)
jits = load_oob(open_with_progress(modeld_pkl_path(usbgpu)) if usbgpu else open_file_chunked(modeld_pkl_path(usbgpu)))
metadata = jits['metadata']
self.input_shapes = metadata['input_shapes']
self.vision_input_names = [k for k in self.input_shapes if 'img' in k]
+1 -2
View File
@@ -108,8 +108,7 @@ class ModelState(ModelStateBase):
def _init_combined(self, pkl_path, cam_w, cam_h, bundle):
cloudlog.warning(f"loading combined pkl: {pkl_path}")
stream = open_with_progress(pkl_path) if self.usbgpu else open_file_chunked(pkl_path)
jits = load_oob(stream)
jits = load_oob(open_with_progress(pkl_path) if self.usbgpu else open_file_chunked(pkl_path))
self.WARP_DEV = 'QCOM' if COMMA_HARDWARE else 'CPU'
self.DEV = 'AMD' if self.usbgpu else self.WARP_DEV