Spaces:
Runtime error
Runtime error
Commit
·
de38bc4
1
Parent(s):
e2beaba
Debug message
Browse files
app.py
CHANGED
@@ -26,12 +26,15 @@ vardecoder_model = AutoModelForCausalLM.from_pretrained(
|
|
26 |
device_map="cuda",
|
27 |
)
|
28 |
print("Loaded vardecoder model successfully.")
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
|
|
|
|
|
|
35 |
|
36 |
make_gradio_client = lambda: Client("https://ejschwartz-resym-field-helper.hf.space/")
|
37 |
|
|
|
26 |
device_map="cuda",
|
27 |
)
|
28 |
print("Loaded vardecoder model successfully.")
|
29 |
+
try:
|
30 |
+
fielddecoder_model = AutoModelForCausalLM.from_pretrained(
|
31 |
+
"ejschwartz/resym-fielddecoder",
|
32 |
+
torch_dtype=torch.bfloat16,
|
33 |
+
low_cpu_mem_usage=True,
|
34 |
+
device_map="cuda",
|
35 |
+
)
|
36 |
+
except Exception as e:
|
37 |
+
print(f"Error loading fielddecoder model: {e}")
|
38 |
|
39 |
make_gradio_client = lambda: Client("https://ejschwartz-resym-field-helper.hf.space/")
|
40 |
|