Text Generation
PEFT
Safetensors
dfurman commited on
Commit
77c90fa
1 Parent(s): dbaf161

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -149,6 +149,10 @@ with torch.cuda.amp.autocast():
149
  print('\n\n', tokenizer.decode(output_tokens[0], skip_special_tokens=True))
150
  ```
151
 
 
 
 
 
152
  ### CUDA Info
153
 
154
  - CUDA Version: 12.0
 
149
  print('\n\n', tokenizer.decode(output_tokens[0], skip_special_tokens=True))
150
  ```
151
 
152
+ ## Reproducibility
153
+
154
+ See attached [Colab Notebook](https://huggingface.co/dfurman/falcon-7b-chat-oasst1/blob/main/finetune_falcon7b_oasst1_with_bnb_peft.ipynb) for the code (and hyperparams) used to train the model.
155
+
156
  ### CUDA Info
157
 
158
  - CUDA Version: 12.0