Skip to content

Commit 47dad6a

Browse files
authored
minor debug (#2908)
1 parent a355793 commit 47dad6a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

gemma-peft.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ We start by downloading the model and the tokenizer. We also include a `BitsAndB
6363

6464
```python
6565
import torch
66+
import os
6667
from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
6768

6869
model_id = "google/gemma-2b"

0 commit comments

Comments
 (0)