vicgalle commited on
Commit
af81a7e
1 Parent(s): e501d43

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -0
README.md CHANGED
@@ -42,5 +42,14 @@ The response is
42
  > There's this famous scene in a diner where... \*Waves hand in front of face\*... where you know what happens, and it's just so \*Blushes bright red\*... realistic and relatable! \*Giggles\* I mean, we've all been there, right? \*Winks\*
43
  > Anyway, it's just such a classic rom-com with great dialogue and a perfect soundtrack! \*Sways dreamily\* I could watch it over and over again. \*Tilts head\* What about you, do you have a favorite film? \*Peeks at you with big, hopeful eyes\* \*Squeals\* I'm dying to know! \*Bounces in seat excitedly\*
44
 
 
 
 
 
 
 
 
 
 
45
 
46
 
 
42
  > There's this famous scene in a diner where... \*Waves hand in front of face\*... where you know what happens, and it's just so \*Blushes bright red\*... realistic and relatable! \*Giggles\* I mean, we've all been there, right? \*Winks\*
43
  > Anyway, it's just such a classic rom-com with great dialogue and a perfect soundtrack! \*Sways dreamily\* I could watch it over and over again. \*Tilts head\* What about you, do you have a favorite film? \*Peeks at you with big, hopeful eyes\* \*Squeals\* I'm dying to know! \*Bounces in seat excitedly\*
44
 
45
+ ## Fine-tuning script
46
+
47
+ The script I used to train this model is available at [this file](./train_human.py). It takes less than one hour in a T4 GPU, and requires the following libs:
48
+
49
+ ```
50
+ peft 0.11.1
51
+ transformers 4.44.0.dev0
52
+ trl 0.9.6
53
+ ``
54
 
55