tranhoangnguyen03
commited on
Commit
•
4ab6e94
1
Parent(s):
7ac0d05
Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ from transformers import AutoTokenizer
|
|
17 |
from auto_fp8 import AutoFP8ForCausalLM, BaseQuantizeConfig
|
18 |
|
19 |
pretrained_model_dir = "UCLA-AGI/Gemma-2-9B-It-SPPO-Iter3"
|
20 |
-
quantized_model_dir = "/
|
21 |
|
22 |
tokenizer = AutoTokenizer.from_pretrained(pretrained_model_dir)
|
23 |
|
|
|
17 |
from auto_fp8 import AutoFP8ForCausalLM, BaseQuantizeConfig
|
18 |
|
19 |
pretrained_model_dir = "UCLA-AGI/Gemma-2-9B-It-SPPO-Iter3"
|
20 |
+
quantized_model_dir = "/quantized/Gemma-2-9B-It-SPPO-Iter3_Q8"
|
21 |
|
22 |
tokenizer = AutoTokenizer.from_pretrained(pretrained_model_dir)
|
23 |
|