diff --git a/ui/media/js/dnd.js b/ui/media/js/dnd.js
index 6e733286..d09fe32f 100644
--- a/ui/media/js/dnd.js
+++ b/ui/media/js/dnd.js
@@ -171,7 +171,7 @@ const TASK_MAPPING = {
         setUI: (use_stable_diffusion_model) => {
             const oldVal = stableDiffusionModelField.value
 
-            use_stable_diffusion_model = getModelPath(use_stable_diffusion_model, ['.ckpt'])
+            use_stable_diffusion_model = getModelPath(use_stable_diffusion_model, ['.ckpt', '.safetensors'])
             stableDiffusionModelField.value = use_stable_diffusion_model
 
             if (!stableDiffusionModelField.value) {