CindyDelage commited on
Commit
c79c66f
·
verified ·
1 Parent(s): 17b7c60

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +2 -2
retriever.py CHANGED
@@ -17,7 +17,7 @@ class FrugalAI_methods(Tool):
17
  ideas.append(quantization())
18
  return ideas
19
 
20
- def pruning(self,method: str):
21
  """
22
  Optimizes models by removing unnecessary components, such as certain weights in a neural network.
23
  This function demonstrates how to apply pruning.
@@ -30,7 +30,7 @@ def pruning(self,method: str):
30
  f"can be increased or decreased depending on the initial number of weights and the complexity of your use case (minimu value: 0, maximum value: 1)."
31
  )
32
 
33
- def quantization(self, method: str):
34
  """
35
  Converts high-precision weights into lower-precision one to reduce cost.
36
  """
 
17
  ideas.append(quantization())
18
  return ideas
19
 
20
+ def pruning():
21
  """
22
  Optimizes models by removing unnecessary components, such as certain weights in a neural network.
23
  This function demonstrates how to apply pruning.
 
30
  f"can be increased or decreased depending on the initial number of weights and the complexity of your use case (minimu value: 0, maximum value: 1)."
31
  )
32
 
33
+ def quantization():
34
  """
35
  Converts high-precision weights into lower-precision one to reduce cost.
36
  """