Update README.md
Browse files
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
|