mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Refactor everything (#3481)

Browse source
This commit is contained in:
oobabooga 2023-08-06 21:49:27 -03:00 • committed by GitHub
parent d4b851bdc8
commit 65aa11890f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
19 changed files with 1306 additions and 1178 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/models.py
View file

@ -1,9 +1,9 @@
import gc
import hashlib
import os
import re
import time
from pathlib import Path
import hashlib
import torch
import transformers
@ -14,7 +14,7 @@ from transformers import (
AutoModelForCausalLM,
AutoModelForSeq2SeqLM,
AutoTokenizer,
BitsAndBytesConfig,
BitsAndBytesConfig
)
import modules.shared as shared