diff --git a/ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py b/ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py
index 6c6aed7e..89ef9e3a 100644
--- a/ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py
+++ b/ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py
@@ -63,9 +63,10 @@
 
 _DEVICE_TYPE = flags.DEFINE_string(
     'device_type',
-    'cpu',
+    None,
     help='The device type of the model. Currently supported: cpu, gpu.',
     default='cpu',
+    required=True,
 )