Spaces:
Runtime error
Runtime error
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
|