Spaces:

ejschwartz
/

resym

Running on Zero

ejschwartz commited on Mar 21

Commit

09910fb

1 Parent(s): d3eb07d

requirements

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,8 +10,8 @@ hf_key = os.environ['HF_TOKEN']
 login(token = hf_key)
 tokenizer = AutoTokenizer.from_pretrained('bigcode/starcoderbase-3b', use_auth_token=hf_key)
-model = AutoModelForCausalLM.from_pretrained(
-    model_path, use_auth_token=hf_key,
     torch_dtype=torch.bfloat16, device_map='auto'
 )

 login(token = hf_key)
 tokenizer = AutoTokenizer.from_pretrained('bigcode/starcoderbase-3b', use_auth_token=hf_key)
+vardecoder_model = AutoModelForCausalLM.from_pretrained(
+    "ejschwartz/resym-vardecoder",
     torch_dtype=torch.bfloat16, device_map='auto'
 )