add access token

app.py

@@ -19,7 +19,7 @@ from transformers import GenerationConfig
 # from modelscope.hub.api import HubApi
 from pydub import AudioSegment
 import os
-# YOUR_ACCESS_TOKEN = os.getenv('YOUR_ACCESS_TOKEN')
+YOUR_ACCESS_TOKEN = os.getenv('YOUR_ACCESS_TOKEN')
 
 # api = HubApi()
 # api.login(YOUR_ACCESS_TOKEN)
@@ -49,7 +49,7 @@ def _get_args():
 
 def _load_model_tokenizer(args):
     tokenizer = AutoTokenizer.from_pretrained(
-        args.checkpoint_path, trust_remote_code=True, resume_download=True,
+        args.checkpoint_path, trust_remote_code=True, resume_download=True, token=YOUR_ACCESS_TOKEN
     )
 
     if args.cpu_only:
@@ -62,9 +62,10 @@ def _load_model_tokenizer(args):
         device_map=device_map,
         trust_remote_code=True,
         resume_download=True,
+        token=YOUR_ACCESS_TOKEN
     ).eval()
     model.generation_config = GenerationConfig.from_pretrained(
-        args.checkpoint_path, trust_remote_code=True, resume_download=True,
+        args.checkpoint_path, trust_remote_code=True, resume_download=True, token=YOUR_ACCESS_TOKEN
     )
 
     return model, tokenizer