spellingdragon commited on
Commit
67b6099
1 Parent(s): acdd9da

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -7,7 +7,7 @@ class EndpointHandler():
7
  def __init__(self, path=""):
8
  #device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
  #torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
10
- #model_id = path
11
 
12
  #model = AutoModelForSpeechSeq2Seq.from_pretrained(
13
  # model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
 
7
  def __init__(self, path=""):
8
  #device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
  #torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
10
+ model_id = "openai/whisper-large-v3"
11
 
12
  #model = AutoModelForSpeechSeq2Seq.from_pretrained(
13
  # model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True