From 9f215523e20256a8a1a96e55ff130605c0e4bad1 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Tue, 6 Jun 2023 07:05:32 -0300 Subject: [PATCH] Remove some unused imports --- download-model.py | 4 ++-- modules/evaluate.py | 1 - modules/text_generation.py | 1 - 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/download-model.py b/download-model.py index 83eab84..f5c2732 100644 --- a/download-model.py +++ b/download-model.py @@ -103,7 +103,7 @@ class ModelDownloader: classifications = [] has_pytorch = False has_pt = False - has_ggml = False + # has_ggml = False has_safetensors = False is_lora = False while True: @@ -148,7 +148,7 @@ class ModelDownloader: has_pt = True classifications.append('pt') elif is_ggml: - has_ggml = True + # has_ggml = True classifications.append('ggml') cursor = base64.b64encode(f'{{"file_name":"{dict[-1]["path"]}"}}'.encode()) + b':50' diff --git a/modules/evaluate.py b/modules/evaluate.py index 3283278..5cd3482 100644 --- a/modules/evaluate.py +++ b/modules/evaluate.py @@ -1,5 +1,4 @@ import datetime -import traceback from pathlib import Path import pandas as pd diff --git a/modules/text_generation.py b/modules/text_generation.py index f4faf4c..6e7c016 100644 --- a/modules/text_generation.py +++ b/modules/text_generation.py @@ -1,7 +1,6 @@ import ast import random import re -import threading import time import traceback