ejschwartz commited on
Commit
903f407
·
1 Parent(s): dc573a3

Try to_empty

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,10 +17,10 @@ huggingface_hub.login(token=hf_key)
17
  tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase-3b")
18
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
19
  "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16
20
- ).to_empty("cuda")
21
  fielddecoder_model = AutoModelForCausalLM.from_pretrained(
22
  "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16
23
- ).to_empty("cuda")
24
 
25
  make_gradio_client = lambda: Client("https://ejschwartz-resym-field-helper.hf.space/")
26
 
 
17
  tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase-3b")
18
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
19
  "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16
20
+ ).to_empty(device="cuda")
21
  fielddecoder_model = AutoModelForCausalLM.from_pretrained(
22
  "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16
23
+ ).to_empty(device="cuda")
24
 
25
  make_gradio_client = lambda: Client("https://ejschwartz-resym-field-helper.hf.space/")
26