Working on it

routers/training.py

@@ -5,6 +5,8 @@ import json
 from routers.donut_evaluate import run_evaluate_donut
 from routers.donut_training import run_training_donut
 import utils
+import torch
+from diffusers import StableDiffusionPipeline
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16)