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",
|