Update models/videochat.py

models/videochat.py

@@ -1,4 +1,5 @@
 import os
+import psutil
 import random
 import logging
 
@@ -115,8 +116,8 @@ class VideoChat(Blip2Base):
         self.llama_tokenizer = LlamaTokenizer.from_pretrained(llama_model_path, use_fast=False, use_auth_token=os.environ["HF_TOKEN"])
         self.llama_tokenizer.pad_token = self.llama_tokenizer.eos_token
 
-        import psutil
-        import os
+        
+        
         print(u'当前进程的内存使用：%.4f GB' % (psutil.Process(os.getpid()).memory_info().rss / 1024 / 1024 / 1024) )
         info = psutil.virtual_memory()
         print( u'电脑总内存：%.4f GB' % (info.total / 1024 / 1024 / 1024) )