mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
125e6e41c6
add model trace and trace obj demo
19 lines
547 B
Python
19 lines
547 B
Python
from rdagent.log.ui.web import WebView, QlibTraceWindow, TraceObjWindow, mock_msg
|
|
from rdagent.log.storage import FileStorage, Message
|
|
from rdagent.core.proposal import Trace
|
|
from pathlib import Path
|
|
import pickle
|
|
|
|
|
|
# show logs folder
|
|
# WebView(QlibTraceWindow(show_common_logs=False, show_llm=False)).display(FileStorage("./log/2024-07-22_03-01-12-021659"))
|
|
|
|
|
|
# load Trace obj
|
|
with Path('./log/step_trace.pkl').open('rb') as f:
|
|
obj = pickle.load(f)
|
|
trace: Trace = obj[-1]
|
|
|
|
# show Trace obj
|
|
# TraceObjWindow().consume_msg(mock_msg(trace))
|