krplatz
/

gemma-med3-9b

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

krplatz commited on Sep 3

Commit

c8186c8

•

1 Parent(s): 948e8bf

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -27,4 +27,4 @@ class EndpointHandler:
 # Function to be called by Hugging Face Inference Toolkit
 def get_pipeline(model_dir):
-    return Gemma2Handler(model_dir)

 # Function to be called by Hugging Face Inference Toolkit
 def get_pipeline(model_dir):
+    return EndpointHandler(model_dir)