Spaces:
Runtime error
Runtime error
GunaKoppula
commited on
Commit
•
b1b49d4
1
Parent(s):
8cebfd9
Update inference/main.py
Browse files- 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="
|
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
|