Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 18:45:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Sort imports according to PEP8 (based on #67)

Browse source
This commit is contained in:
oobabooga 2023-02-10 15:40:03 -03:00
parent 96d56d4f3c
commit 219366342b
2 changed files with 25 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
download-model.py
View file

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

32
server.py
View file

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