Tonic commited on
Commit
3eda1f6
1 Parent(s): 7f2687d

GemmaForCausalLM import error fix

Browse files

adding the GemmaForCausalLM as import for transformers as the instructions to use the provided code do not describe how to import gemma otherwise.

Files changed (1) hide show
  1. README.md +1 -2
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 gemma.modeling_gemma import GemmaForCausalLM
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