GunaKoppula commited on
Commit
b1b49d4
1 Parent(s): 8cebfd9

Update inference/main.py

Browse files
Files changed (1) hide show
  1. inference/main.py +2 -2
inference/main.py CHANGED
@@ -72,7 +72,7 @@ class WhisperWithProjection:
72
  class MultiModalPhi2:
73
  def __init__(
74
  self,
75
- modelname_or_path="RaviNaik/Llava-Phi2",
76
  temperature=0.2,
77
  max_new_tokens=1024,
78
  device="cuda:0",
@@ -223,4 +223,4 @@ class MultiModalPhi2:
223
  if outputs.endswith(stop_str):
224
  outputs = outputs[: -len(stop_str)]
225
  outputs = outputs.strip()
226
- return outputs
 
72
  class MultiModalPhi2:
73
  def __init__(
74
  self,
75
+ modelname_or_path="GunaKoppula/Llava-Phi2",
76
  temperature=0.2,
77
  max_new_tokens=1024,
78
  device="cuda:0",
 
223
  if outputs.endswith(stop_str):
224
  outputs = outputs[: -len(stop_str)]
225
  outputs = outputs.strip()
226
+ return outputs