zzc0208 commited on
Commit
90968b7
1 Parent(s): 2317e3f

Update useapi.py

Browse files
Files changed (1) hide show
  1. useapi.py +1 -1
useapi.py CHANGED
@@ -64,7 +64,7 @@ def get_models():
64
  if model_info['model'] in ['gemma-2-9b', 'llama-3.1-8b']:
65
  weight = 12 # Assign a low weight to reduce their frequency
66
  else:
67
- weight = int(math.ceil(25 / model_info['price'] + 0.5))
68
  self.models.extend([model_info['model']] * weight)
69
 
70
  model_storage = ModelStorage()
 
64
  if model_info['model'] in ['gemma-2-9b', 'llama-3.1-8b']:
65
  weight = 12 # Assign a low weight to reduce their frequency
66
  else:
67
+ weight = int(math.ceil(25 / (model_info['price'] + 0.5)))
68
  self.models.extend([model_info['model']] * weight)
69
 
70
  model_storage = ModelStorage()