jadehardouin
commited on
Commit
•
fd49c3f
1
Parent(s):
47a336c
Update models.py
Browse files
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
|
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
|