Fix an issue in device type flag.

PiperOrigin-RevId: 692563283
google-ai-edge · Nov 3, 2024 · 29aa17f · 29aa17f
1 parent bd09407
commit 29aa17f
Showing 1 changed file with 1 addition and 3 deletions.
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,10 +63,8 @@
 
 _DEVICE_TYPE = flags.DEFINE_string(
     'device_type',
-    None,
+    'cpu',
     help='The device type of the model. Currently supported: cpu, gpu.',
-    default='cpu',
-    required=True,
 )