Spaces:
Running
on
Zero
Running
on
Zero
0.38 testing shuffle
Browse files
app.py
CHANGED
@@ -73,8 +73,8 @@ def apply_chat_template(messages, add_generation_prompt=False):
|
|
73 |
|
74 |
@spaces.GPU()
|
75 |
def generate_both(system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens=2048, temperature=0.2, top_p=0.9, repetition_penalty=1.1):
|
76 |
-
|
77 |
try:
|
|
|
78 |
text_streamer_a = TextIteratorStreamer(tokenizer_a, skip_prompt=True)
|
79 |
text_streamer_b = TextIteratorStreamer(tokenizer_b, skip_prompt=True)
|
80 |
|
|
|
73 |
|
74 |
@spaces.GPU()
|
75 |
def generate_both(system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens=2048, temperature=0.2, top_p=0.9, repetition_penalty=1.1):
|
|
|
76 |
try:
|
77 |
+
logging.debug('Pass just once')
|
78 |
text_streamer_a = TextIteratorStreamer(tokenizer_a, skip_prompt=True)
|
79 |
text_streamer_b = TextIteratorStreamer(tokenizer_b, skip_prompt=True)
|
80 |
|