dadni commited on
Commit
018a028
·
verified ·
1 Parent(s): a0f36db

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -17,7 +17,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
17
  from huggingface_hub import hf_hub_download
18
 
19
  # 1. Download the quantized weights
20
- weights_path = hf_hub_download(repo_id="dadni/GPT2-Medium-2Bit-Demo", filename="gpt2-sym-matmulfree.pt")
21
 
22
  # 2. Load the base architecture and tokenizer
23
  model_id = "gpt2-medium"
 
17
  from huggingface_hub import hf_hub_download
18
 
19
  # 1. Download the quantized weights
20
+ weights_path = hf_hub_download(repo_id="dadni/GPT2-Medium-2Bit-Demo", filename="gpt2-2bit-matmulfree.pt")
21
 
22
  # 2. Load the base architecture and tokenizer
23
  model_id = "gpt2-medium"