Spaces:
Running
on
Zero
Running
on
Zero
added cache to prepare_model
Browse files- sotopia_generate.py +1 -1
sotopia_generate.py
CHANGED
|
@@ -81,7 +81,7 @@ def generate_action(
|
|
| 81 |
# print(e)
|
| 82 |
# return AgentAction(action_type="none", argument="")
|
| 83 |
|
| 84 |
-
|
| 85 |
def prepare_model(model_name):
|
| 86 |
compute_type = torch.float16
|
| 87 |
|
|
|
|
| 81 |
# print(e)
|
| 82 |
# return AgentAction(action_type="none", argument="")
|
| 83 |
|
| 84 |
+
@cache
|
| 85 |
def prepare_model(model_name):
|
| 86 |
compute_type = torch.float16
|
| 87 |
|