enable pylint

cli/modules/ffmpeg.py

@@ -8,6 +8,7 @@ import subprocess
 import pathlib
 import argparse
 import filetype
+
 from util import log, Map
 
 

cli/modules/grid.py

@@ -13,6 +13,7 @@ import filetype
 from PIL import Image, ImageDraw, ImageFont
 from util import log
 
+
 params = None
 
 

cli/modules/ideas.py

@@ -6,12 +6,16 @@ model from: <https://huggingface.co/FredZhang7/distilgpt2-stable-diffusion-v2>
 
 import logging
 import argparse
+
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
+
 from util import log
 
+
 tokenizer = None
 model = None
 
+
 def prompt(text: str, temp: float = 0.9, top: int = 8, penalty: float = 1.2, alpha: float = 0.6, num: int = 5, length: int = 80):
     global tokenizer, model # pylint: disable=global-statement
     if tokenizer is None:

cli/modules/interrogate.py

@@ -8,8 +8,10 @@ import base64
 import sys
 import os
 import asyncio
+
 import filetype
 from PIL import Image
+
 from util import log, Map
 import sdapi as sdapi
 

cli/modules/losschart.py

@@ -6,12 +6,15 @@ import sys
 import json
 import pathlib
 import logging
+
 import torch
 import numpy as np
 from PIL import Image, ImageFont, ImageDraw
 from matplotlib import pyplot as plt
+
 from util import log, Map
 
+
 def settings(logdir: str, name: str):
     filename = os.path.join(logdir, name, 'settings.json')
     with open(filename, 'r', encoding='utf-8') as f:

cli/modules/lossrate.py

@@ -5,14 +5,17 @@ auto-generate learn-rate
 import io
 import math
 import logging
+
 import numpy as np
 from PIL import Image, ImageFont, ImageDraw
 from matplotlib import pyplot as plt
+
 from util import log, Map
 
 
 loss_types = ['linear', 'log', 'linalg', 'power']
 
+
 def gen_steps(steps, step):
     return [x for x in range(1, steps + step) if x % step == 0]
 

cli/modules/process.py

@@ -8,9 +8,11 @@ import io
 import shutil
 import base64
 import pathlib
+
 import numpy as np
 import mediapipe as mp
 from PIL import Image, ImageOps
+
 from util import log, Map
 from sdapi import postsync
 

cli/modules/promptist.py

@@ -5,9 +5,12 @@ use microsoft promptist to beautify prompt
 """
 
 import sys
+
 from transformers import AutoModelForCausalLM, AutoTokenizer
+
 from util import log
 
+
 def load_prompter():
     model = AutoModelForCausalLM.from_pretrained("microsoft/Promptist") # pylint: disable=redefined-outer-name
     tokenizer = AutoTokenizer.from_pretrained("gpt2") # pylint: disable=redefined-outer-name

cli/modules/sdapi.py

@@ -4,21 +4,22 @@ helper methods that creates HTTP session with managed connection pool
 provides async HTTP get/post methods and several helper methods
 """
 
+import aiohttp
 import asyncio
 import logging
+import requests
 import sys
 
-import aiohttp
-import requests
 from util import Map, log
 
+
 sd_url = "http://127.0.0.1:7860" # automatic1111 api url root
 use_session = True
 timeout = aiohttp.ClientTimeout(total = None, sock_connect = 10, sock_read = None) # default value is 5 minutes, we need longer for training
-
 sess = None
 quiet = False
 
+
 async def result(req):
     if req.status != 200:
         if not quiet:

cli/modules/util.py

@@ -5,9 +5,11 @@ generic helper methods
 
 import logging
 
+
 logging.basicConfig(level = logging.INFO, format = '%(asctime)s %(levelname)s: %(message)s')
 log = logging.getLogger("sd")
 
+
 class Map(dict):
     def __init__(self, *args, **kwargs):
         super(Map, self).__init__(*args, **kwargs)
@@ -45,5 +47,6 @@ class Map(dict):
         super(Map, self).__delitem__(key)
         del self.__dict__[key]
 
+
 if __name__ == "__main__":
     pass