Update app.py

app.py

@@ -7,8 +7,9 @@ from diffusers import StableDiffusionPipeline, DDIMScheduler
 move_cache()
 
 model_path = "slymnyldrm/dreambooth_usecase_weights"
+commandline_args = os.environ.get('COMMANDLINE_ARGS', "--skip-torch-cuda-test --no-half")
 
-pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float32)
+pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16)
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 
 g_cuda = torch.Generator()