clean

Author: hjh0119

cookbook/legacy/grpo/gsm8k.py

@@ -62,7 +62,6 @@
 # SwanLab experiment tracking
 USE_SWANLAB = bool(int(os.environ.get('USE_SWANLAB', '1')))
 if USE_SWANLAB:
-    os.environ['SWANLAB_API_KEY'] = '3hVJrk0veNB2NCm72UdJg'
     import swanlab
     swanlab.login(api_key=os.environ['SWANLAB_API_KEY'], save=True)
     swanlab.init(project="twinkle-gsm8k", config={

cookbook/legacy/grpo/lora_backup.py

@@ -15,7 +15,7 @@
 from twinkle.dataset import Dataset, DatasetMeta
 from twinkle.model import TransformersModel
 from twinkle.processor import InputProcessor
-from twinkle.sampler import VLLMSampler
+from twinkle.sampler import vLLMSampler
 from twinkle.template import Template
 from twinkle.metric import CompletionRewardMetric
 
@@ -134,7 +134,7 @@ def main():
     model.set_processor(InputProcessor, adapter_name=ADAPTER_NAME)
     model.set_template('Template', model_id=MODEL_ID, adapter_name=ADAPTER_NAME)
 
-    sampler = VLLMSampler(
+    sampler = vLLMSampler(
         model_id=MODEL_ID,
         engine_args={
             'load_format': 'dummy',