Hugging Face's logo

Spaces:
ejschwartz
/
resym
Running on Zero

App Files Community
ejschwartz commited on
Commit
97087c6
·
1 Parent(s): 4ee21c2

use gpu

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,10 +15,10 @@ huggingface_hub.login(token=hf_key)
15
 
16
  tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase-3b")
17
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
18
- "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map="auto"
19
  )
20
  fielddecoder_model = AutoModelForCausalLM.from_pretrained(
21
- "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16, device_map="auto"
22
  )
23
 
24
  gradio_client = Client("https://ejschwartz-resym-field-helper.hf.space/")
 
15
 
16
  tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase-3b")
17
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
18
+ "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map={"": 0}
19
  )
20
  fielddecoder_model = AutoModelForCausalLM.from_pretrained(
21
+ "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16, device_map={"": 0}
22
  )
23
 
24
  gradio_client = Client("https://ejschwartz-resym-field-helper.hf.space/")