jadehardouin commited on
Commit
fd49c3f
1 Parent(s): 47a336c

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -274,7 +274,7 @@ class ModelPage:
274
  if current_model == model.get_name():
275
  model_args = args[begin:begin+model_n_args]
276
  cost_per_input_token, cost_per_output_token, labor_cost = model.compute_cost_per_token(*model_args)
277
- model_tco = cost_per_input_token * current_input_tokens.value + cost_per_output_token * current_output_tokens.value
278
  latency = model.get_latency()
279
 
280
  return model_tco, latency, labor_cost
 
274
  if current_model == model.get_name():
275
  model_args = args[begin:begin+model_n_args]
276
  cost_per_input_token, cost_per_output_token, labor_cost = model.compute_cost_per_token(*model_args)
277
+ model_tco = cost_per_input_token * current_input_tokens + cost_per_output_token * current_output_tokens
278
  latency = model.get_latency()
279
 
280
  return model_tco, latency, labor_cost