jerryzh168 commited on
Commit
265080b
·
verified ·
1 Parent(s): a204b4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -46,7 +46,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
46
 
47
  # Push to hub
48
  USER_ID = "YOUR_USER_ID"
49
- save_to = f"{USER_ID}/{model_id}-float8dq"
 
50
  quantized_model.push_to_hub(save_to, safe_serialization=False)
51
  tokenizer.push_to_hub(save_to)
52
 
 
46
 
47
  # Push to hub
48
  USER_ID = "YOUR_USER_ID"
49
+ MODEL_NAME = model_id.split("/")[-1]
50
+ save_to = f"{USER_ID}/{MODEL_NAME}-float8dq"
51
  quantized_model.push_to_hub(save_to, safe_serialization=False)
52
  tokenizer.push_to_hub(save_to)
53