Spaces:
Sleeping
Sleeping
Commit
·
711ba71
1
Parent(s):
fd26237
increase duration
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ client.add(collection_name="recipes",
|
|
| 40 |
model_name = "LeoLM/leo-hessianai-70b-chat"
|
| 41 |
|
| 42 |
last_messages = []
|
| 43 |
-
@spaces.GPU(duration=
|
| 44 |
def load_model():
|
| 45 |
ankerbot_model = AutoModelForCausalLM.from_pretrained(
|
| 46 |
model_name,
|
|
@@ -60,7 +60,7 @@ def load_model():
|
|
| 60 |
|
| 61 |
_model_cache = None
|
| 62 |
|
| 63 |
-
@spaces.GPU
|
| 64 |
def get_model():
|
| 65 |
global _model_cache
|
| 66 |
print(f"Model: {_model_cache}")
|
|
|
|
| 40 |
model_name = "LeoLM/leo-hessianai-70b-chat"
|
| 41 |
|
| 42 |
last_messages = []
|
| 43 |
+
@spaces.GPU(duration=180)
|
| 44 |
def load_model():
|
| 45 |
ankerbot_model = AutoModelForCausalLM.from_pretrained(
|
| 46 |
model_name,
|
|
|
|
| 60 |
|
| 61 |
_model_cache = None
|
| 62 |
|
| 63 |
+
@spaces.GPU(duration=120)
|
| 64 |
def get_model():
|
| 65 |
global _model_cache
|
| 66 |
print(f"Model: {_model_cache}")
|