mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-08-23 13:13:44 +08:00
simplify
This commit is contained in:
@@ -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]
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user