Spaces:

TIGER-Lab
/

MAmmoTH2

Running on Zero

aaabiao commited on May 24, 2024

Commit

8f70d80

verified ·

1 Parent(s): 4cbf1bf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,13 @@ from typing import Iterator
 import gradio as gr
 import spaces
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer, StoppingCriteriaList, StoppingCriteriaSub
 MAX_MAX_NEW_TOKENS = 2048
 DEFAULT_MAX_NEW_TOKENS = 1024

 import gradio as gr
 import spaces
 import torch
+from transformers import (
+    AutoModelForCausalLM,
+    AutoTokenizer,
+    StoppingCriteria,
+    StoppingCriteriaList,
+    TextIteratorStreamer,
+)
 MAX_MAX_NEW_TOKENS = 2048
 DEFAULT_MAX_NEW_TOKENS = 1024