jackbond2024 commited on
Commit
416da8e
1 Parent(s): 62f8d75

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. trans_cli_demo.py +1 -1
  2. trans_web_demo.py +1 -0
trans_cli_demo.py CHANGED
@@ -18,7 +18,7 @@ from threading import Thread
18
  from transformers import AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer, AutoModel
19
 
20
  MODEL_PATH = os.environ.get('MODEL_PATH', 'THUDM/glm-4-9b-chat')
21
-
22
  print("MODEL_PATH: " + MODEL_PATH)
23
 
24
  ## If use peft model.
 
18
  from transformers import AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer, AutoModel
19
 
20
  MODEL_PATH = os.environ.get('MODEL_PATH', 'THUDM/glm-4-9b-chat')
21
+ MODEL_PATH = "/Users/zmac/Documents/opensrc/llms/GLM-4/models"
22
  print("MODEL_PATH: " + MODEL_PATH)
23
 
24
  ## If use peft model.
trans_web_demo.py CHANGED
@@ -29,6 +29,7 @@ TokenizerType = Union[PreTrainedTokenizer, PreTrainedTokenizerFast]
29
 
30
  MODEL_PATH = os.environ.get('MODEL_PATH', 'THUDM/glm-4-9b-chat')
31
  #MODEL_PATH = "/Users/zmac/.cache/huggingface/hub/models--THUDM--glm-4-9b-chat/snapshots/04419001bc63e05e70991ade6da1f91c4aeec278"
 
32
  TOKENIZER_PATH = os.environ.get("TOKENIZER_PATH", MODEL_PATH)
33
 
34
 
 
29
 
30
  MODEL_PATH = os.environ.get('MODEL_PATH', 'THUDM/glm-4-9b-chat')
31
  #MODEL_PATH = "/Users/zmac/.cache/huggingface/hub/models--THUDM--glm-4-9b-chat/snapshots/04419001bc63e05e70991ade6da1f91c4aeec278"
32
+ MODEL_PATH = "/Users/zmac/Documents/opensrc/llms/GLM-4/models"
33
  TOKENIZER_PATH = os.environ.get("TOKENIZER_PATH", MODEL_PATH)
34
 
35