소스 검색

Sort imports according to PEP8 (based on #67)

oobabooga 3 년 전
부모
커밋
219366342b
2개의 변경된 파일23개의 추가작업 그리고 19개의 파일을 삭제
  1. 6 6
      download-model.py
  2. 17 13
      server.py

+ 6 - 6
download-model.py

@@ -5,15 +5,15 @@ Example:
 python download-model.py facebook/opt-1.3b
 
 '''
-
-import requests 
-from bs4 import BeautifulSoup 
+import argparse
 import multiprocessing
-import tqdm
+import re
 import sys
-import argparse
 from pathlib import Path
-import re
+
+import requests
+import tqdm
+from bs4 import BeautifulSoup
 
 parser = argparse.ArgumentParser()
 parser.add_argument('MODEL', type=str)

+ 17 - 13
server.py

@@ -1,26 +1,30 @@
-import re
+import argparse
+import base64
+import copy
 import gc
-import time
 import glob
-import torch
-import argparse
-import json
 import io
-import base64
-import sys
+import json
 import os
+import re
+import sys
+import time
+import warnings
 from datetime import datetime
 from pathlib import Path
-from PIL import Image
-import copy
+
 import gradio as gr
-import warnings
-from tqdm import tqdm
+import torch
 import transformers
-from transformers import AutoTokenizer, AutoModelForCausalLM, AutoConfig
+from PIL import Image
+from tqdm import tqdm
+from transformers import AutoConfig
+from transformers import AutoModelForCausalLM
+from transformers import AutoTokenizer
+
 from modules.html_generator import *
-from modules.ui import *
 from modules.stopping_criteria import _SentinelTokenStoppingCriteria
+from modules.ui import *
 
 transformers.logging.set_verbosity_error()