woods-today commited on
Commit
430abad
1 Parent(s): 7eb427a

Working on it

Browse files
__pycache__/config.cpython-311.pyc ADDED
Binary file (1.45 kB). View file
 
routers/__pycache__/__init__.cpython-311.pyc ADDED
Binary file (184 Bytes). View file
 
routers/__pycache__/inference.cpython-311.pyc ADDED
Binary file (4.51 kB). View file
 
routers/training.py CHANGED
@@ -6,6 +6,8 @@ from routers.donut_evaluate import run_evaluate_donut
6
  from routers.donut_training import run_training_donut
7
  import utils
8
 
 
 
9
 
10
  router = APIRouter()
11
 
 
6
  from routers.donut_training import run_training_donut
7
  import utils
8
 
9
+ pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16)
10
+
11
 
12
  router = APIRouter()
13