Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|