Nanobit's picture
Fix set mem_id for inference and refactor
974dc00