add todo

siliconflow · Nov 6, 2024 · 9319cca · 9319cca
1 parent b13521c
commit 9319cca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onediff_diffusers_extensions/examples/flux/text_to_image_flux.py b/onediff_diffusers_extensions/examples/flux/text_to_image_flux.py
@@ -259,7 +259,7 @@ def main():
             if get_gpu_memory() > 24:
                 _ = quantize(pipe.transformer, **quantize_config)
             else:
-                # for gpu with little memory, such as 4090
+                # (TODO:support 4090) for gpu with little memory, such as 4090
                 if hasattr(pipe, "transformer"):
                     pipe.transformer = pipe.transformer.to("cuda")
                     _ = quantize(pipe.transformer, **quantize_config)