Ahmet Kaan Sever commited on
Commit
211b909
·
1 Parent(s): e8c3b4b

Commented more peft related stuff for testing

Browse files
Files changed (2) hide show
  1. svc/router.py +2 -1
  2. 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