mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Accelerate: Remove HAS_ACCELERATE
Accelerate has been a dependency for a while, and as such we probably shouldn't be lugging around code that assumes it isn't present.
This commit is contained in:
7
utils.py
7
utils.py
@@ -9,7 +9,6 @@ import requests
|
||||
import requests.adapters
|
||||
import time
|
||||
import breakmodel
|
||||
from transformers import __version__ as transformers_version
|
||||
from transformers import PreTrainedModel
|
||||
import packaging.version
|
||||
from tqdm.auto import tqdm
|
||||
@@ -21,12 +20,6 @@ import packaging.version
|
||||
from pathlib import Path
|
||||
from typing import List, Optional
|
||||
|
||||
HAS_ACCELERATE = packaging.version.parse(transformers_version) >= packaging.version.parse("4.20.0.dev0")
|
||||
try:
|
||||
import accelerate
|
||||
except ImportError:
|
||||
HAS_ACCELERATE = False
|
||||
|
||||
koboldai_vars = None
|
||||
args = None
|
||||
num_shards: Optional[int] = None
|
||||
|
Reference in New Issue
Block a user