Spaces:

nvidia
/

Eagle2-Demo

Running on Zero

Zhiding commited on 14 days ago

Commit

79b84d6

1 Parent(s): 2c8a4a7

update

Files changed (1) hide show

eagle_vl/serve/inference.py CHANGED Viewed

@@ -69,7 +69,7 @@ def load_model(model_path: str = "nvidia/Eagle-2.5-8B"):
         model, processor = load_model_from_nv(model_path)
     except Exception as e:
         logger.error(f"Failed to load model from HF, trying to load from eagle: {e}")
-        model, processor = load_model_from_eagle(model_path)
     return model, processor
 class StoppingCriteriaSub(StoppingCriteria):

         model, processor = load_model_from_nv(model_path)
     except Exception as e:
         logger.error(f"Failed to load model from HF, trying to load from eagle: {e}")
+        model, processor = load_model_from_eagle()
     return model, processor
 class StoppingCriteriaSub(StoppingCriteria):