Feature/add unet effnetb7

Emojich.md

@@ -51,15 +51,24 @@ show(pil_images, 4)
 ```
 ![](./pics/emojich/emoji-Donald.png)
 
+### Super Resolution:
+```python
+from rudalle.pipelines import super_resolution
+from rudalle import get_realesrgan
+
+device = 'cuda'
+realesrgan = get_realesrgan('x4', device=device)
+sr_images = super_resolution(pil_images, realesrgan)
+```
+
 ### Converting to Telegram Stickers format (512x512 RGBA)
-Also, don't forget you can use super resolution before :)
 ```python
 from rudalle.pipelines import convert_emoji_to_rgba, show_rgba
 from rudalle import get_emojich_unet
 
 device = 'cuda'
-emojich_unet = get_emojich_unet('unet_effnetb5').to(device)
-rgba_images = convert_emoji_to_rgba(pil_images, emojich_unet,  device=device)
+emojich_unet = get_emojich_unet('unet_effnetb7').to(device)
+rgba_images = convert_emoji_to_rgba(sr_images, emojich_unet,  device=device)
 for rgba_image in rgba_images:
     show_rgba(rgba_image);
 ```

README.md

@@ -7,7 +7,7 @@
 [![pre-commit.ci status](https://results.pre-commit.ci/badge/github/sberbank-ai/ru-dalle/master.svg)](https://results.pre-commit.ci/latest/github/sberbank-ai/ru-dalle/master)
 
 ```
-pip install rudalle==0.2.1
+pip install rudalle==0.3.0
 ```
 ### 🤗 HF Models:
 [ruDALL-E Malevich (XL)](https://huggingface.co/sberbank-ai/rudalle-Malevich) \

rudalle/__init__.py

@@ -24,4 +24,4 @@
     'image_prompts',
 ]
 
-__version__ = '0.2.1'
+__version__ = '0.3.0'

rudalle/emojich_unet/__init__.py

@@ -2,7 +2,6 @@
 import os
 
 import torch
-import segmentation_models_pytorch as smp
 from huggingface_hub import hf_hub_url, cached_download
 
 
@@ -12,12 +11,24 @@
         repo_id='sberbank-ai/rudalle-Emojich',
         filename='pytorch_model.bin',
     ),
+    'unet_effnetb7': dict(
+        encoder_name='efficientnet-b7',
+        repo_id='sberbank-ai/rudalle-Emojich',
+        filename='pytorch_model.bin',
+    ),
 }
 
 
 def get_emojich_unet(name, cache_dir='/tmp/rudalle'):
     assert name in MODELS
     config = MODELS[name]
+    try:
+        import segmentation_models_pytorch as smp
+    except ImportError:
+        import logging
+        logging.warning('If you would like to use emojich_unet, you should reinstall timm package:'
+                        '"pip install timm==0.4.12"')
+        return
     model = smp.Unet(
         encoder_name=config['encoder_name'],
         encoder_weights=None,

rudalle/pipelines.py

@@ -118,7 +118,7 @@ def show(pil_images, nrow=4, size=14, save_dir=None, show=True):
         plt.show()
 
 
-def convert_emoji_to_rgba(pil_images, emojich_unet,  device='cpu', bs=4):
+def convert_emoji_to_rgba(pil_images, emojich_unet,  device='cpu', bs=1):
     final_images = []
     with torch.no_grad():
         for chunk in more_itertools.chunked(pil_images, bs):