Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
49a522b
1
Parent(s):
d43763e
do we need this
Browse files
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"
|
|
|
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"
|