Update README.md

README.md

@@ -17,9 +17,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 if __name__ == "__main__":
     PROMPT = "def square_sum(xs):\n    return sum(x * x for x in xs)\n\nsquare_sum([1, 2, 3])\n"
-    tok = AutoTokenizer.from_pretrained("circuit-sparsity", trust_remote_code=True)
+    tok = AutoTokenizer.from_pretrained("openai/circuit-sparsity", trust_remote_code=True)
     model = AutoModelForCausalLM.from_pretrained(
-        "circuit-sparsity",
+        "openai/circuit-sparsity",
         trust_remote_code=True,
         torch_dtype="auto",
     )