purbeshmitra
/

vanillaGRPO

Text Generation

Model card Files Files and versions

purbeshmitra commited on Jul 7

Commit

12d19a4

·

verified ·

1 Parent(s): ba38ea0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,8 +44,8 @@ Our results are shown below:
 ## Usage
 ```python
-from peft import PeftModel
 from transformers import AutoModelForCausalLM
 base_model = AutoModelForCausalLM.from_pretrained("unsloth/qwen2.5-1.5b-instruct-unsloth-bnb-4bit")
 model = PeftModel.from_pretrained(base_model, "purbeshmitra/vanillaGRPO")

 ## Usage
 ```python
 from transformers import AutoModelForCausalLM
+from peft import PeftModel
 base_model = AutoModelForCausalLM.from_pretrained("unsloth/qwen2.5-1.5b-instruct-unsloth-bnb-4bit")
 model = PeftModel.from_pretrained(base_model, "purbeshmitra/vanillaGRPO")