From 27438c0e5e757101e7158003e92e0c4dabbcf5c4 Mon Sep 17 00:00:00 2001
From: Haoliang Zhang <haoliang@google.com>
Date: Tue, 29 Oct 2024 12:31:44 -0700
Subject: [PATCH] fix convert code.

PiperOrigin-RevId: 691120287
---
 .../generative/examples/stable_diffusion/convert_to_tflite.py  | 3 ++-
 1 file changed, 2 insertions(+), 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
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,
 )