Mini-Gemini

Runtime error

wcy1122 commited on Apr 10, 2024

Commit

e97863d

1 Parent(s): 528ddd1

update code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -203,7 +203,6 @@ def generate(state, imagebox, textbox, image_process_mode, gen_image, temperatur
     torch.cuda.empty_cache()
-@spaces.GPU
 def add_text(state, imagebox, textbox, image_process_mode, gen_image):
     if state is None:
         state = conv_templates[conv_mode].copy()

     torch.cuda.empty_cache()
 def add_text(state, imagebox, textbox, image_process_mode, gen_image):
     if state is None:
         state = conv_templates[conv_mode].copy()