autoparse correct vram total when available

Comfy-Org · Aug 26, 2024 · fd6c11a · fd6c11a
1 parent 5ca346b
commit fd6c11a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/components/UsageGraph.tsx b/components/UsageGraph.tsx
@@ -1,11 +1,16 @@
 import React from 'react';
 import { LineChart, Line, XAxis, YAxis, CartesianGrid, Tooltip, ResponsiveContainer } from 'recharts';
 
-const UsageGraph = ({ data, maxMemory }) => {
-    const parsedData = Object.entries(data).map(([time, mem]) => ({
-        time: parseFloat(time.split(' ')[0]),
-        mem: parseInt((mem as string).split(' ')[0])
-    })).sort((a, b) => a.time - b.time);
+const UsageGraph = ({ data }) => {
+    const maxMemory = data["total"] ? parseInt(data["total"].split(' ')[0]) : undefined;
+
+    const parsedData = Object.entries(data)
+        .filter(([key]) => key !== "total")
+        .map(([time, mem]) => ({
+            time: parseFloat(time.split(' ')[0]),
+            mem: parseInt((mem as string).split(' ')[0])
+        }))
+        .sort((a, b) => a.time - b.time);
 
     const maxYValue = maxMemory || Math.max(...parsedData.map(item => item.mem));
 

diff --git a/pages/workflow/[id].tsx b/pages/workflow/[id].tsx
@@ -158,7 +158,7 @@ function WorkflowResultDetail() {
                     VRAM Usage Over Time
                 </h5>
                 <div className="w-full">
-                    <UsageGraph data={workflowResult?.machine_stats?.vram_time_series || {}} maxMemory={workflowResult?.peak_vram} />
+                    <UsageGraph data={workflowResult?.machine_stats?.vram_time_series || {}} />
                 </div>
             </Card>
         </div>