update README

README.md

@@ -74,14 +74,15 @@ IQuest-Coder-V1 uses custom modeling code via Hugging Face's auto_map feature. W
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "IQuest/IQuest-Coder-V1-40B-Instruct"
+model_name = "IQuestLab/IQuest-Coder-V1-40B-Instruct"
 
 # Load the tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     torch_dtype="auto",
-    device_map="auto"
+    device_map="cuda:0",
+    trust_remote_code=True,
 )
 
 # Prepare the input