Update README.md

README.md

@@ -19,12 +19,11 @@ You can load it using our branch of the `dictionary_learning` library:
 !pip install git+https://github.com/jkminder/dictionary_learning
 from dictionary_learning import CrossCoder
 from nnsight import LanguageModel
-from dlabutils import model_path
 import torch as th
 
 crosscoder = CrossCoder.from_pretrained("Butanium/gemma-2-2b-crosscoder-l13-mu4.1e-02-lr1e-04", from_hub=True)
-gemma_2 = LanguageModel(model_path("google/gemma-2-2b"), device_map="cuda:0")
-gemma_2_it = LanguageModel(model_path("google/gemma-2-2b-it"), device_map="cuda:1")
+gemma_2 = LanguageModel("google/gemma-2-2b", device_map="cuda:0")
+gemma_2_it = LanguageModel("google/gemma-2-2b-it", device_map="cuda:1")
 prompt = "quick fox brown"
 
 with gemma_2.trace(prompt):