diff --git a/engine/commands/ps_cmd.cc b/engine/commands/ps_cmd.cc index fab19b9b3..2b2c8dc8d 100644 --- a/engine/commands/ps_cmd.cc +++ b/engine/commands/ps_cmd.cc @@ -4,6 +4,7 @@ #include #include #include "nlohmann/json.hpp" +#include "utils/format_utils.h" #include "utils/logging_utils.h" #include "utils/string_utils.h" @@ -52,8 +53,8 @@ void PsCmd::PrintModelStatusList( table.add_row({ model_status.model, model_status.engine, - model_status.ram, - model_status.vram, + format_utils::BytesToHumanReadable(model_status.ram), + format_utils::BytesToHumanReadable(model_status.vram), string_utils::FormatTimeElapsed(model_status.start_time), }); } diff --git a/engine/commands/ps_cmd.h b/engine/commands/ps_cmd.h index c48d3d97f..3f2ef266f 100644 --- a/engine/commands/ps_cmd.h +++ b/engine/commands/ps_cmd.h @@ -1,5 +1,6 @@ #pragma once +#include #include #include @@ -8,9 +9,9 @@ namespace commands { struct ModelLoadedStatus { std::string engine; std::string model; - std::string ram; + uint64_t ram; uint64_t start_time; - std::string vram; + uint64_t vram; }; class PsCmd {