GemmaForCausalLM import error fix
Browse filesadding the GemmaForCausalLM as import for transformers as the instructions to use the provided code do not describe how to import gemma otherwise.
README.md
CHANGED
@@ -46,8 +46,7 @@ Octopus-V2-2B, an advanced open-source language model with 2 billion parameters,
|
|
46 |
|
47 |
You can run the model on a GPU using the following code.
|
48 |
```python
|
49 |
-
from
|
50 |
-
from transformers import AutoTokenizer
|
51 |
import torch
|
52 |
import time
|
53 |
|
|
|
46 |
|
47 |
You can run the model on a GPU using the following code.
|
48 |
```python
|
49 |
+
from transformers import AutoTokenizer, GemmaForCausalLM
|
|
|
50 |
import torch
|
51 |
import time
|
52 |
|