Upload handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -11,7 +11,7 @@ class EndpointHandler():
|
|
11 |
model_id,
|
12 |
model_file=model_file,
|
13 |
model_type="mistral",
|
14 |
-
gpu_layers=
|
15 |
**config
|
16 |
)
|
17 |
self.pipeline = llm
|
|
|
11 |
model_id,
|
12 |
model_file=model_file,
|
13 |
model_type="mistral",
|
14 |
+
gpu_layers=50,#50 #110
|
15 |
**config
|
16 |
)
|
17 |
self.pipeline = llm
|