Spaces:

Enderchef
/

SuperBench-Eval

Sleeping

Enderchef commited on Jun 25

Commit

bddb36d

verified ·

1 Parent(s): 468784f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -72,6 +72,7 @@ def load_model(model_id):
         model = AutoModelForCausalLM.from_pretrained(
             model_id,
             token=HF_TOKEN,
             torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32
         ).to("cuda" if torch.cuda.is_available() else "cpu")

         model = AutoModelForCausalLM.from_pretrained(
             model_id,
             token=HF_TOKEN,
+            trust_remote_code=True,
             torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32
         ).to("cuda" if torch.cuda.is_available() else "cpu")