Update README.md

README.md

@@ -118,8 +118,8 @@ device = torch.device("cuda:1" if torch.cuda.is_available() else "CPU")
 
 # load model
 # IF you cannot connect to huggingface.co, you can download the repo and set from_pretrained path as the loacl dir path, replacing "yuxindu/segvol"
-clip_tokenizer = AutoTokenizer.from_pretrained("./segvol")
-model = AutoModel.from_pretrained("./segvol", trust_remote_code=True, test_mode=False)
+clip_tokenizer = AutoTokenizer.from_pretrained("yuxindu/segvol")
+model = AutoModel.from_pretrained("yuxindu/segvol", trust_remote_code=True, test_mode=False)
 model.model.text_encoder.tokenizer = clip_tokenizer
 model.train()
 model.to(device)