Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
7553bf5
1
Parent(s):
a28e277
use of local model
Browse files- util/utils.py +1 -1
util/utils.py
CHANGED
|
@@ -75,7 +75,7 @@ def load_model_tok(model_name="gpt2-xl"):
|
|
| 75 |
|
| 76 |
elif model_name == 'llama-3-8b':
|
| 77 |
|
| 78 |
-
model = "meta-llama/Meta-Llama-3-8B"
|
| 79 |
tok = AutoTokenizer.from_pretrained(model)
|
| 80 |
model = AutoModelForCausalLM.from_pretrained(
|
| 81 |
model,
|
|
|
|
| 75 |
|
| 76 |
elif model_name == 'llama-3-8b':
|
| 77 |
|
| 78 |
+
model = "/data/meta-llama/Meta-Llama-3-8B"
|
| 79 |
tok = AutoTokenizer.from_pretrained(model)
|
| 80 |
model = AutoModelForCausalLM.from_pretrained(
|
| 81 |
model,
|