Spaces:
Sleeping
Sleeping
Ahmet Kaan Sever
commited on
Commit
·
211b909
1
Parent(s):
e8c3b4b
Commented more peft related stuff for testing
Browse files- svc/router.py +2 -1
- svc/schemas.py +3 -3
svc/router.py
CHANGED
@@ -97,7 +97,8 @@ def deep_eval_suite(request: DeepEvalSuiteRequest):
|
|
97 |
def run_in_background():
|
98 |
try:
|
99 |
torch.cuda.empty_cache()
|
100 |
-
des = DeepEvalTaskManager(request.model_name, request.tasks, request.dtype, request.weight, request.base_model)
|
|
|
101 |
|
102 |
start_time = time()
|
103 |
results = des.run_tasks()
|
|
|
97 |
def run_in_background():
|
98 |
try:
|
99 |
torch.cuda.empty_cache()
|
100 |
+
#des = DeepEvalTaskManager(request.model_name, request.tasks, request.dtype, request.weight, request.base_model)
|
101 |
+
des = DeepEvalTaskManager(request.model_name, request.tasks)
|
102 |
|
103 |
start_time = time()
|
104 |
results = des.run_tasks()
|
svc/schemas.py
CHANGED
@@ -29,9 +29,9 @@ class LMHarnessTaskRequest(BaseModel):
|
|
29 |
|
30 |
class DeepEvalSuiteRequest(BaseModel):
|
31 |
model_name: str
|
32 |
-
dtype: str
|
33 |
-
weight: str
|
34 |
-
base_model: str
|
35 |
tasks: Optional[List[str]] = None
|
36 |
|
37 |
|
|
|
29 |
|
30 |
class DeepEvalSuiteRequest(BaseModel):
|
31 |
model_name: str
|
32 |
+
# dtype: str
|
33 |
+
# weight: str
|
34 |
+
# base_model: str
|
35 |
tasks: Optional[List[str]] = None
|
36 |
|
37 |
|