Update src/demo.py

src/demo.py

@@ -20,9 +20,9 @@ type2dataset = {
     "text2re-hard": load_dataset('3B-Group/ConvRe', "en-text2re", token=TOKEN, split="prompt3")
 }
 
-model_id = "TheBloke/Llama-2-7b-Chat-GPTQ"
-tokenizer = AutoTokenizer.from_pretrained(model_id, token=TOKEN, use_fast=True)
-model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, token=TOKEN, trust_remote_code=False, revision="main").eval()
+model_id = "meta-llama/Llama-2-7b-chat-hf"
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=TOKEN)
+model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, token=TOKEN).eval()
 
 # type2dataset = {}