5to9 commited on
Commit
c250107
·
1 Parent(s): 58cb5ad

0.25 trying to decorate model load funcs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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: