Hugging Face's logo

Spaces:
ejschwartz
/
resym
Running on Zero

App Files Community
ejschwartz commited on
Commit
84b4a86
·
1 Parent(s): acaa2a2

no device map?

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -15,11 +15,11 @@ 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={"": 0}
19
- ).to("cuda")
20
  fielddecoder_model = AutoModelForCausalLM.from_pretrained(
21
- "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16, device_map={"": 0}
22
- ).to("cuda")
23
 
24
  gradio_client = Client("https://ejschwartz-resym-field-helper.hf.space/")
25
 
 
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/")
25