Do not call forward directly
This commit is contained in:
2
main.py
2
main.py
@@ -3,7 +3,7 @@ from src import CodexAgent, CodexAgentConfig
|
||||
codex_agent = CodexAgent(CodexAgentConfig())
|
||||
|
||||
def main():
|
||||
codex_agent.push_to_hub("darinkishore/codex-agent", with_code=True)
|
||||
codex_agent.push_to_hub("darinkishore/codex-agent", with_code=True, commit_message="Do not call forward directly")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -34,4 +34,4 @@ class CodexAgent(PrecompiledAgent):
|
||||
)
|
||||
|
||||
def forward(self, **kwargs) -> Prediction:
|
||||
return self.codex_module.forward(**kwargs)
|
||||
return self.codex_module(**kwargs)
|
||||
Reference in New Issue
Block a user