GemmaForCausalLM import error fix (#3)

- GemmaForCausalLM import error fix (3eda1f6e8363425f4045262d9957ff7ff401e9a5)


Co-authored-by: Joseph Pollack <[email protected]>

README.md

@@ -46,8 +46,7 @@ Octopus-V2-2B, an advanced open-source language model with 2 billion parameters,
 
 You can run the model on a GPU using the following code. 
 ```python
-from gemma.modeling_gemma import GemmaForCausalLM
-from transformers import AutoTokenizer
+from transformers import AutoTokenizer, GemmaForCausalLM
 import torch
 import time