tranhoangnguyen03 commited on
Commit
4ab6e94
1 Parent(s): 7ac0d05

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = "/content/Gemma-2-9B-It-SPPO-Iter3_Q8"
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