diff --git a/.gitignore b/.gitignore
index 2db7461..14decca 100644
--- a/.gitignore
+++ b/.gitignore
@@ -166,4 +166,4 @@ runs/
 jupyters/custom_*
 
 *logs/
-.DS_store
\ No newline at end of file
+.DS_store
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index d63bec2..5ef24ea 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,31 +1,22 @@
 repos:
 -   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v2.2.3
+    rev: v4.0.1
     hooks:
     -   id: check-docstring-first
-        stages:
-        - commit
-        - push
     -   id: check-merge-conflict
         stages:
         - push
     -   id: double-quote-string-fixer
-        stages:
-        - commit
-        - push
+    -   id: end-of-file-fixer
     -   id: fix-encoding-pragma
-        stages:
-        - commit
-        - push
+    -   id: mixed-line-ending
+    -   id: trailing-whitespace
+-   repo: https://github.com/pycqa/flake8
+    rev: "4.0.1"
+    hooks:
     -   id: flake8
         args: ['--config=setup.cfg']
-        stages:
-        - commit
-        - push
 -   repo: https://github.com/pre-commit/mirrors-autopep8
-    rev: v1.4.4
+    rev: v1.5.7
     hooks:
     -   id: autopep8
-        stages:
-        - commit
-        - push
diff --git a/README.md b/README.md
index 171fa77..1fcc650 100644
--- a/README.md
+++ b/README.md
@@ -56,7 +56,7 @@ for top_k, top_p, images_num in [
     (512, 0.97, 3),
     (384, 0.96, 3),
     (256, 0.95, 3),
-    (128, 0.95, 3), 
+    (128, 0.95, 3),
 ]:
     _pil_images, _scores = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, top_p=top_p)
     pil_images += _pil_images
@@ -97,6 +97,6 @@ skyes = [red_sky, sunny_sky, cloudy_sky, night_sky]
 
 - [@neverix](https://www.kaggle.com/neverix) thanks a lot for contributing for speed up of inference
 - [@Igor Pavlov](https://github.com/boomb0om) trained model and prepared code with [super-resolution](https://github.com/boomb0om/Real-ESRGAN-colab)
-- [@oriBetelgeuse](https://github.com/oriBetelgeuse) thanks a lot for easy API of generation using image prompt 
-- [@Alex Wortega](https://github.com/AlexWortega) created first FREE version colab notebook with fine-tuning [ruDALL-E Malevich (XL)](https://huggingface.co/sberbank-ai/rudalle-Malevich) on sneakers domain 💪 
+- [@oriBetelgeuse](https://github.com/oriBetelgeuse) thanks a lot for easy API of generation using image prompt
+- [@Alex Wortega](https://github.com/AlexWortega) created first FREE version colab notebook with fine-tuning [ruDALL-E Malevich (XL)](https://huggingface.co/sberbank-ai/rudalle-Malevich) on sneakers domain 💪
 - [@Anton Lozhkov](https://github.com/anton-l) Integrated to [Huggingface Spaces](https://huggingface.co/spaces) with [Gradio](https://github.com/gradio-app/gradio), see [here](https://huggingface.co/spaces/anton-l/rudall-e)
diff --git a/requirements.txt b/requirements.txt
index c358382..43ab492 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,4 @@ omegaconf>=2.0.0
 einops~=0.3.2
 torch
 torchvision
-matplotlib
\ No newline at end of file
+matplotlib