diff --git a/setup_files/txt2img.py b/setup_files/txt2img.py
index 377ecdb..0a4907f 100644
--- a/setup_files/txt2img.py
+++ b/setup_files/txt2img.py
@@ -138,7 +138,7 @@ class StableDiffusion:
         """
         Runs the Stable Diffusion pipeline on the given prompt and outputs images.
         """
-        import pillow_avif  # noqa
+        import pillow_avif  # noqa: F401
         import torch
 
         max_embeddings_multiples = self._count_token(p=prompt, n=n_prompt)