Update agent.py
Browse files
agent.py
CHANGED
@@ -53,11 +53,6 @@ import sys
|
|
53 |
|
54 |
import weave
|
55 |
weave.init("gaia-llamaindex-agents")
|
56 |
-
from llama_index.core import set_global_handler
|
57 |
-
|
58 |
-
# Set Weave as the global callback handler
|
59 |
-
set_global_handler("weave")
|
60 |
-
|
61 |
def get_max_memory_config(max_memory_per_gpu):
|
62 |
"""Generate max_memory config for available GPUs"""
|
63 |
if torch.cuda.is_available():
|
|
|
53 |
|
54 |
import weave
|
55 |
weave.init("gaia-llamaindex-agents")
|
|
|
|
|
|
|
|
|
|
|
56 |
def get_max_memory_config(max_memory_per_gpu):
|
57 |
"""Generate max_memory config for available GPUs"""
|
58 |
if torch.cuda.is_available():
|