Spaces:

ejschwartz
/

test

Runtime error

ejschwartz commited on Jun 9

Commit

e2beaba

1 Parent(s): d3fb139

Debug message

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,6 +25,7 @@ vardecoder_model = AutoModelForCausalLM.from_pretrained(
     low_cpu_mem_usage=True,
     device_map="cuda",
 )
 fielddecoder_model = AutoModelForCausalLM.from_pretrained(
     "ejschwartz/resym-fielddecoder",
     torch_dtype=torch.bfloat16,

     low_cpu_mem_usage=True,
     device_map="cuda",
 )
+print("Loaded vardecoder model successfully.")
 fielddecoder_model = AutoModelForCausalLM.from_pretrained(
     "ejschwartz/resym-fielddecoder",
     torch_dtype=torch.bfloat16,