Explorar o código

Minor formatting changes

oobabooga %!s(int64=2) %!d(string=hai) anos
pai
achega
cb169d0834
Modificáronse 2 ficheiros con 2 adicións e 6 borrados
  1. 0 4
      modules/text_generation.py
  2. 2 2
      server.py

+ 0 - 4
modules/text_generation.py

@@ -63,8 +63,6 @@ def generate_softprompt_input_tensors(input_ids):
     return inputs_embeds, filler_input_ids
 
 # Removes empty replies from gpt4chan outputs
-
-
 def fix_gpt4chan(s):
     for i in range(10):
         s = re.sub("--- [0-9]*\n>>[0-9]*\n---", "---", s)
@@ -73,8 +71,6 @@ def fix_gpt4chan(s):
     return s
 
 # Fix the LaTeX equations in galactica
-
-
 def fix_galactica(s):
     s = s.replace(r'\[', r'$')
     s = s.replace(r'\]', r'$')

+ 2 - 2
server.py

@@ -398,13 +398,13 @@ def create_interface():
             reload_inputs = [shared.gradio[k] for k in ['name1', 'name2', 'Chat mode']]
 
             gen_events.append(shared.gradio['Generate'].click(
-                lambda x : (x, ''), shared.gradio['textbox'], [shared.gradio['Chat input'], shared.gradio['textbox']], show_progress=False).then(
+                lambda x: (x, ''), shared.gradio['textbox'], [shared.gradio['Chat input'], shared.gradio['textbox']], show_progress=False).then(
                 chat.cai_chatbot_wrapper, shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream).then(
                 lambda: chat.save_history(timestamp=False), [], [], show_progress=False)
             )
 
             gen_events.append(shared.gradio['textbox'].submit(
-                lambda x : (x, ''), shared.gradio['textbox'], [shared.gradio['Chat input'], shared.gradio['textbox']], show_progress=False).then(
+                lambda x: (x, ''), shared.gradio['textbox'], [shared.gradio['Chat input'], shared.gradio['textbox']], show_progress=False).then(
                 chat.cai_chatbot_wrapper, shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream).then(
                 lambda: chat.save_history(timestamp=False), [], [], show_progress=False)
             )