Spaces:
Running
on
Zero
Running
on
Zero
0.25 trying to decorate model load funcs
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def apply_chat_template(messages, add_generation_prompt=False):
|
|
54 |
|
55 |
return pharia_template
|
56 |
|
57 |
-
|
58 |
def load_model_a(model_id):
|
59 |
global tokenizer_a, model_a, model_id_a
|
60 |
try:
|
@@ -70,7 +70,7 @@ def load_model_a(model_id):
|
|
70 |
logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
|
71 |
return gr.update(label=model_id)
|
72 |
|
73 |
-
|
74 |
def load_model_b(model_id):
|
75 |
global tokenizer_b, model_b, model_id_b
|
76 |
try:
|
|
|
54 |
|
55 |
return pharia_template
|
56 |
|
57 |
+
@spaces.GPU()
|
58 |
def load_model_a(model_id):
|
59 |
global tokenizer_a, model_a, model_id_a
|
60 |
try:
|
|
|
70 |
logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
|
71 |
return gr.update(label=model_id)
|
72 |
|
73 |
+
@spaces.GPU()
|
74 |
def load_model_b(model_id):
|
75 |
global tokenizer_b, model_b, model_id_b
|
76 |
try:
|