Update app.py

app.py

@@ -10,7 +10,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "tensorart/stable-diffusion-3.5-large-TurboX"
 
 if torch.cuda.is_available():
-    torch_dtype = torch.bfloat16
+    torch_dtype = torch.float16
 else:
     torch_dtype = torch.float32
 
@@ -63,8 +63,8 @@ css = """
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(" # [Stable Diffusion 3.5 Large (8B)](https://huggingface.co/stabilityai/stable-diffusion-3.5-large)")
-        gr.Markdown("[Learn more](https://stability.ai/news/introducing-stable-diffusion-3-5) about the Stable Diffusion 3.5 series. Try on [Stability AI API](https://platform.stability.ai/docs/api-reference#tag/Generate/paths/~1v2beta~1stable-image~1generate~1sd3/post), or [download model](https://huggingface.co/stabilityai/stable-diffusion-3.5-large) to run locally with ComfyUI or diffusers.")
+        gr.Markdown(" # [TensorArt Stable Diffusion 3.5 Large TurboX)](https://huggingface.co/tensorart/stable-diffusion-3.5-large-TurboX)")
+        gr.Markdown("8-step distilled turbo model")
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",