revert torch_size so is 16 on cuda

app.py

@@ -197,8 +197,8 @@ def on_embeddings_changed_update_plot(embeddings_b64):
             width=embeddings.shape[0])
 
 device = torch.device("mps" if torch.backends.mps.is_available() else "cuda:0" if torch.cuda.is_available() else "cpu")
-# torch_size = torch.float16 if device == ('cuda') else torch.float32
-torch_size = torch.float32
+torch_size = torch.float16 if device == ('cuda') else torch.float32
+# torch_size = torch.float32
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,
     custom_pipeline="pipeline.py",