use device

This commit is contained in:
Josh Hawkins 2025-04-06 22:55:05 -05:00
parent 0866a60a94
commit c29d42d766

View File

@ -108,7 +108,9 @@ class EmbeddingMaintainer(threading.Thread):
# model runners to share between realtime and post processors # model runners to share between realtime and post processors
if self.config.lpr.enabled: if self.config.lpr.enabled:
lpr_model_runner = LicensePlateModelRunner(self.requestor, device="AUTO") lpr_model_runner = LicensePlateModelRunner(
self.requestor, device=self.config.lpr.device
)
# realtime processors # realtime processors
self.realtime_processors: list[RealTimeProcessorApi] = [] self.realtime_processors: list[RealTimeProcessorApi] = []