Lint

extensions/openai/completions.py

@@ -1,14 +1,14 @@
 import copy
-import time
 import json
+import time
 from collections import deque
 
 import tiktoken
+from pydantic import ValidationError
 
 from extensions.openai.errors import InvalidRequestError
-from extensions.openai.utils import debug_msg, getToolCallId, parseToolCall
 from extensions.openai.typing import ToolDefinition
-from pydantic import ValidationError
+from extensions.openai.utils import debug_msg, getToolCallId, parseToolCall
 from modules import shared
 from modules.chat import (
     generate_chat_prompt,

extensions/openai/utils.py

@@ -1,9 +1,9 @@
 import base64
-import os
-import time
 import json
+import os
 import random
 import re
+import time
 import traceback
 from typing import Callable, Optional
 

extensions/superboogav2/chromadb.py

@@ -1,10 +1,11 @@
 import math
 import random
 import threading
-import torch
+
 import chromadb
 import numpy as np
 import posthog
+import torch
 from chromadb.config import Settings
 from chromadb.utils import embedding_functions
 

modules/tensorrt_llm.py

@@ -1,15 +1,15 @@
 from pathlib import Path
 
-import torch
-
 import tensorrt_llm
+import torch
+from tensorrt_llm.runtime import ModelRunner, ModelRunnerCpp
+
 from modules import shared
 from modules.logging_colors import logger
 from modules.text_generation import (
     get_max_prompt_length,
     get_reply_from_output_ids
 )
-from tensorrt_llm.runtime import ModelRunner, ModelRunnerCpp
 
 
 class TensorRTLLMModel: