okezieowen commited on
Commit
94dbb00
·
verified ·
1 Parent(s): 4900414

Update handler.py

Browse files

Changing the handler class from EndpointHandlerV2 to EndpointHandler

Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -9,7 +9,7 @@ import traceback
9
  from transformers import AutoModelForCausalLM, AutoTokenizer
10
  from snac import SNAC
11
 
12
- class EndpointHandlerV2:
13
  def __init__(self, path=""):
14
 
15
  # Delimiter tokens as defined in Orpheus' vocabulary
 
9
  from transformers import AutoModelForCausalLM, AutoTokenizer
10
  from snac import SNAC
11
 
12
+ class EndpointHandler:
13
  def __init__(self, path=""):
14
 
15
  # Delimiter tokens as defined in Orpheus' vocabulary