tianleliphoebe commited on
Commit
1f17b02
1 Parent(s): b8cc9c8

Update serve/vote_utils.py

Browse files
Files changed (1) hide show
  1. serve/vote_utils.py +4 -2
serve/vote_utils.py CHANGED
@@ -764,8 +764,10 @@ def generate_igm_annoy_museum(gen_func, state0, state1, model_name0, model_name1
764
  ip = get_ip(request)
765
  igm_logger.info(f"generate. ip: {ip}")
766
  start_tstamp = time.time()
767
- model_name0 = re.sub(r"### Model A: ", "", model_name0)
768
- model_name1 = re.sub(r"### Model B: ", "", model_name1)
 
 
769
  generated_image0, generated_image1, model_name0, model_name1, text = gen_func(model_name0, model_name1)
770
  state0.prompt = text
771
  state1.prompt = text
 
764
  ip = get_ip(request)
765
  igm_logger.info(f"generate. ip: {ip}")
766
  start_tstamp = time.time()
767
+ # model_name0 = re.sub(r"### Model A: ", "", model_name0)
768
+ # model_name1 = re.sub(r"### Model B: ", "", model_name1)
769
+ model_name0 = ""
770
+ model_name1 = ""
771
  generated_image0, generated_image1, model_name0, model_name1, text = gen_func(model_name0, model_name1)
772
  state0.prompt = text
773
  state1.prompt = text