marvinscham/masterthesis-playground
1
0
Fork 0
mirror of https://github.com/marvinscham/masterthesis-playground.git synced 2026-03-22 08:22:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add helper stuff for figures, cleanup

Browse Source
This commit is contained in:
Marvin Scham
2026-02-20 01:56:28 +01:00
parent 28823dc0b5
commit 101bd81ca1
20 changed files with 1862 additions and 1164 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
raft/train_mistral_raft.py
View File

@@ -8,8 +8,6 @@ from peft import LoraConfig
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
from trl import SFTConfig, SFTTrainer
## Usage: python train_mistral_raft.py --train_jsonl out/raft_train.jsonl --out_dir out/mistral_balitwin_lora
def main():
ap = argparse.ArgumentParser()