Update boomerang.py

boomerang.py

@@ -11,9 +11,11 @@ from diffusers import StableDiffusionPipeline
 from diffusers.models.unet_2d_condition import UNet2DConditionOutput
 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
 
-# Use fp16 if running on Google Colab
-#pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=os.environ.get('HF_TOKEN_SD'))  
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=os.environ.get('HF_TOKEN_SD'))  
+COLAB = False
+if COLAB:
+  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=os.environ.get('HF_TOKEN_SD')) 
+else: 
+  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=os.environ.get('HF_TOKEN_SD'))  
 
 #pipe = pipe.to('cuda')