Working on it

endpoints.py

@@ -3,6 +3,11 @@ from fastapi.middleware.cors import CORSMiddleware
 from routers import inference, training
 from huggingface_hub import login
 from config import settings
+import torch
+from diffusers import StableDiffusionPipeline
+
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16)
+
 
 login(settings.huggingface_key)