Add application file

sd3_pipeline.py

@@ -69,6 +69,7 @@ EXAMPLE_DOC_STRING = """
         ```
 """
 
+@torch.cuda.amp.autocast(dtype=torch.float32)
 def optimized_scale(positive_flat, negative_flat):
 
     # Calculate dot production