ejschwartz commited on
Commit
de38bc4
·
1 Parent(s): e2beaba

Debug message

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -26,12 +26,15 @@ vardecoder_model = AutoModelForCausalLM.from_pretrained(
26
  device_map="cuda",
27
  )
28
  print("Loaded vardecoder model successfully.")
29
- fielddecoder_model = AutoModelForCausalLM.from_pretrained(
30
- "ejschwartz/resym-fielddecoder",
31
- torch_dtype=torch.bfloat16,
32
- low_cpu_mem_usage=True,
33
- device_map="cuda",
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