Spaces:

asdc
/

temporal_expression_normalization

Runtime error

asdc commited on May 20

Commit

c588f84

verified ·

1 Parent(s): d5bcccc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
 base_model = AutoModelForCausalLM.from_pretrained(
     BASE_MODEL,
     torch_dtype=torch.float16,
-    device_map="auto"
     token=API_KEY
 )

 base_model = AutoModelForCausalLM.from_pretrained(
     BASE_MODEL,
     torch_dtype=torch.float16,
+    device_map="auto",
     token=API_KEY
 )