sujal7102003 commited on
Commit
0cd0aaf
·
verified ·
1 Parent(s): 6fd4ad5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -70,7 +70,7 @@ async def process_menu(request: Request, file: UploadFile = File(...)):
70
 
71
  # Stable Diffusion
72
  pipe = StableDiffusionPipeline.from_pretrained(
73
- "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16
74
  ).to("cpu")
75
 
76
  cards = []
 
70
 
71
  # Stable Diffusion
72
  pipe = StableDiffusionPipeline.from_pretrained(
73
+ "runwayml/stable-diffusion-v1-5"
74
  ).to("cpu")
75
 
76
  cards = []