dar-tau commited on
Commit
2f28571
1 Parent(s): 5ae12bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ def initialize_gpu():
43
 
44
  def reset_model(model_name, *extra_components, with_extra_components=True):
45
  # extract model info
 
 
46
  model_args = deepcopy(model_info[model_name])
47
  model_path = model_args.pop('model_path')
48
  global_state.original_prompt_template = model_args.pop('original_prompt_template')
 
43
 
44
  def reset_model(model_name, *extra_components, with_extra_components=True):
45
  # extract model info
46
+ print('HELLO!!')
47
+
48
  model_args = deepcopy(model_info[model_name])
49
  model_path = model_args.pop('model_path')
50
  global_state.original_prompt_template = model_args.pop('original_prompt_template')