Update app.py

app.py

@@ -257,7 +257,7 @@ with gr.Blocks() as demo:
                         train_batch_size,
                         gradient_accumulation_steps,
                 ):
-                    try:
+                    if 1:
                         run_optimization = partial(
                             run_main,                 
                             num_tokens=int(num_tokens),
@@ -271,8 +271,8 @@ with gr.Blocks() as demo:
                         run_optimization()
                         print('finish')
                         return "Optimization finished!"
-                    except:
-                        return "CUDA out of memory, use a smaller batch size or try another picture."
+                    #except:
+                    #    return "CUDA out of memory, use a smaller batch size or try another picture."
 
                 
                 def immediate_update():