diff --git a/Cargo.toml b/Cargo.toml
index f78722c..89905f2 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -6,7 +6,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [features]
-default = ["wgpu-backend"]
+default = ["torch-backend"]
 torch-backend = ["burn-tch"]
 wgpu-backend = ["burn-wgpu"]
 
diff --git a/README.md b/README.md
index d5e3fdf..37c0c75 100644
--- a/README.md
+++ b/README.md
@@ -22,17 +22,17 @@ wget https://huggingface.co/Gadersd/Stable-Diffusion-Burn/resolve/main/V1/SDv1-4
 
 ### Step 2: Run the Sample Binary
 
-Invoke the sample binary provided in the rust code. By default, wgpu is used which requires a gpu with at least 10 GB of VRAM (will be lower in the future), but torch can be used with the `torch-backend` feature and can run on a 6 GB gpu.
+Invoke the sample binary provided in the rust code. By default, torch is used. The WGPU backend is unstable for SD but may work well in the future as burn-wpu is optimized.
 
 ```bash
-# wgpu (NEEDS >= 10 GB VRAM)
-# Arguments: <model_type(burn or dump)> <model> <unconditional_guidance_scale> <n_diffusion_steps> <prompt> <output_image>
-cargo run --release --bin sample burn SDv1-4 7.5 20 "An ancient mossy stone." img
-
 # torch (at least 6 GB VRAM, possibly less)
 export TORCH_CUDA_VERSION=cu113
 # Arguments: <model_type(burn or dump)> <model> <unconditional_guidance_scale> <n_diffusion_steps> <prompt> <output_image>
-cargo run --release --features torch-backend --bin sample burn SDv1-4 7.5 20 "An ancient mossy stone." img
+cargo run --release --bin sample burn SDv1-4 7.5 20 "An ancient mossy stone." img
+
+# wgpu (UNSTABLE)
+# Arguments: <model_type(burn or dump)> <model> <unconditional_guidance_scale> <n_diffusion_steps> <prompt> <output_image>
+cargo run --release --features wgpu-backend --bin sample burn SDv1-4 7.5 20 "An ancient mossy stone." img
 ```
 
 This command will generate an image according to the provided prompt, which will be saved as 'img0.png'.