Spaces:
Sleeping
Sleeping
Update inference/main.py
Browse files- inference/main.py +1 -1
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",
|
|
|
|
| 72 |
class MultiModalPhi2:
|
| 73 |
def __init__(
|
| 74 |
self,
|
| 75 |
+
modelname_or_path="Navyabhat/Llava-Phi2",
|
| 76 |
temperature=0.2,
|
| 77 |
max_new_tokens=1024,
|
| 78 |
device="cuda:0",
|