Final_Assignment_Template

Configuration error

Coool2 commited on 14 days ago

Commit

b1f5ba8

1 Parent(s): b2c7d9d

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -68,7 +68,7 @@ weave.init("gaia-llamaindex-agents")
 from transformers import get_max_memory
-max_mem = get_max_memory(0.9)  # 90% of each device's memory
 # Initialize models based on API availability
 def initialize_models(use_api_mode=False):

 from transformers import get_max_memory
+max_mem = get_max_memory(0.75)  # 75% of each device's memory
 # Initialize models based on API availability
 def initialize_models(use_api_mode=False):