DoctorSlimm commited on
Commit
da98440
1 Parent(s): 442299b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
9
 
10
 
11
  # Load the model and processor
12
-
13
- MODEL_PATH = "THUDM/cogvlm2-llama3-chat-19B"
14
 
15
  os.environ['HF_HUB_ENABLE_HF_TRANSFER'] = '1'
16
  MODEL_PATH = snapshot_download(MODEL_PATH)
 
9
 
10
 
11
  # Load the model and processor
12
+ # MODEL_PATH = "THUDM/cogvlm2-llama3-chat-19B"
13
+ MODEL_PATH = "THUDM/cogvlm2-llama3-chat-19B-int4"
14
 
15
  os.environ['HF_HUB_ENABLE_HF_TRANSFER'] = '1'
16
  MODEL_PATH = snapshot_download(MODEL_PATH)