Spaces:

mithril-security
/

TCO_calculator

Runtime error

jadehardouin commited on Aug 8, 2023

Commit

406259e

1 Parent(s): 2a3c4ea

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -168,7 +168,7 @@ class OpenSourceLlama2Model(BaseTCOModel):
                                    visible=False)
     def compute_cost_per_token(self, vm_cost_per_hour, tokens_per_second, maxed_out, used):
-        cost_per_token = vm_cost_per_hour / (tokens_per_second * 3600 * maxed_out * used)
         return cost_per_token
 class OpenSourceDIY(BaseTCOModel):
@@ -211,7 +211,7 @@ class OpenSourceDIY(BaseTCOModel):
                                    visible=False)
     def compute_cost_per_token(self, vm_cost_per_hour, tokens_per_second, maxed_out, used):
-        cost_per_token = vm_cost_per_hour / (tokens_per_second * 3600 * maxed_out * used)
         return cost_per_token
 class CohereModel(BaseTCOModel):

                                    visible=False)
     def compute_cost_per_token(self, vm_cost_per_hour, tokens_per_second, maxed_out, used):
+        cost_per_token = vm_cost_per_hour * 10000 / (tokens_per_second * 3600 * maxed_out * used)
         return cost_per_token
 class OpenSourceDIY(BaseTCOModel):
                                    visible=False)
     def compute_cost_per_token(self, vm_cost_per_hour, tokens_per_second, maxed_out, used):
+        cost_per_token = vm_cost_per_hour * 10000 / (tokens_per_second * 3600 * maxed_out * used)
         return cost_per_token
 class CohereModel(BaseTCOModel):