Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Tom Aarsen
commited on
Commit
•
578a44d
1
Parent(s):
2e647d9
Avoid "0.41000000000000003" GB of memory usage
Browse files- refresh.py +2 -0
refresh.py
CHANGED
@@ -715,6 +715,8 @@ def load_results(data_path: str) -> list | dict | pd.DataFrame | str | None:
|
|
715 |
df = pd.read_json(data_path, orient="records", lines=True)
|
716 |
if "index" in df.columns:
|
717 |
df = df.set_index("index")
|
|
|
|
|
718 |
return df
|
719 |
|
720 |
else:
|
|
|
715 |
df = pd.read_json(data_path, orient="records", lines=True)
|
716 |
if "index" in df.columns:
|
717 |
df = df.set_index("index")
|
718 |
+
if "Memory Usage (GB, fp32)" in df.columns:
|
719 |
+
df["Memory Usage (GB, fp32)"] = df["Memory Usage (GB, fp32)"].map(lambda value: round(value, 2) if isinstance(value, float) else value)
|
720 |
return df
|
721 |
|
722 |
else:
|