ejschwartz commited on
Commit
49a522b
·
1 Parent(s): d43763e

do we need this

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,7 +10,8 @@ hf_key = os.environ["HF_TOKEN"]
10
  huggingface_hub.login(token=hf_key)
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(
13
- "bigcode/starcoderbase-3b", use_auth_token=hf_key
 
14
  )
15
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
16
  "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map="auto"
 
10
  huggingface_hub.login(token=hf_key)
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(
13
+ "bigcode/starcoderbase-3b"
14
+ #, use_auth_token=hf_key
15
  )
16
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
17
  "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map="auto"