fix convert code.

PiperOrigin-RevId: 691120287
google-ai-edge · Oct 29, 2024 · 27438c0 · 27438c0
1 parent 1c1dcb5
commit 27438c0
Showing 1 changed file with 2 additions and 1 deletion.
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
@@ -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,
 )