kkkkk

inference.py

@@ -360,7 +360,7 @@ class Chat:
 
 if __name__ == '__main__':
     # model_path = "/mnt/petrelfs/zhangqinglong/Documents/Husky/work_dirs/husky_v3/EmbodiedGPT/pretrain_0727"
-    model_path = "/mnt/petrelfs/share_data/gvembodied/workdirs/align_new_myyf"
+    model_path = "./"
     device = "cuda" if torch.cuda.is_available() else "cpu"
     chat = Chat(model_path, device=device, num_gpus=1, max_new_tokens=1024, load_8bit=False)
 

requirements.txt

@@ -6,7 +6,8 @@ torchaudio==2.0.2
 transformers==4.28.1 
 sentence_transformers==2.2.2
 decord==0.6.0
-huggingface_hub>=0.16.4,<0.18
+huggingface_hub==0.13.3
+tokenizers==0.13.3
 peft==0.4.0
 Pillow
 einops