modaic/code-generator-trl-grpo
3
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(no commit message)

Browse Source
This commit is contained in:
Farouk Adeleke
2025-12-24 00:33:38 -08:00
parent 76508b0062
commit 3982ab9745
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -4,8 +4,8 @@ from modaic import PrecompiledProgram, PrecompiledConfig
class CodeGeneratorConfig(PrecompiledConfig):
model : str = "openai//models/checkpoint-5"
api_base : str = "https://modaic-ai--grpo-demo-serve.modal.run/v1"
max_tokens : int = 1024
temperature : float = 0.7
max_tokens : int = 10000
temperature : float = 2
class CodeGeneration(dspy.Signature):
@@ -32,6 +32,6 @@ class CodeGenerator(PrecompiledProgram):
code_generator = CodeGenerator(CodeGeneratorConfig())
print(code_generator(query="Write a python function that returns the sum of two numbers.").code)
code_generator.push_to_hub("modaic/code-generator-trl-grpo", with_code=True, tag="v1")
code_generator.push_to_hub("modaic/code-generator-trl-grpo", with_code=True, tag="v2")