mirror of https://github.com/leafspark/AutoGGUF
122 lines
4.6 KiB
Python
122 lines
4.6 KiB
Python
import locale
|
|
import os
|
|
import platform
|
|
import shutil
|
|
import socket
|
|
import time
|
|
from datetime import datetime
|
|
|
|
import psutil
|
|
from PySide6.QtCore import QRegularExpression, Signal
|
|
from PySide6.QtGui import QDoubleValidator, QIntValidator, QRegularExpressionValidator
|
|
from PySide6.QtWidgets import QComboBox, QHBoxLayout, QLineEdit, QPushButton, QWidget
|
|
|
|
|
|
class KVOverrideEntry(QWidget):
|
|
deleted = Signal(QWidget)
|
|
|
|
def __init__(self, parent=None) -> None:
|
|
super().__init__(parent)
|
|
layout = QHBoxLayout(self)
|
|
layout.setContentsMargins(0, 0, 0, 0)
|
|
|
|
self.key_input = QLineEdit()
|
|
self.key_input.setPlaceholderText("Key")
|
|
# Set validator for key input (letters and dots only)
|
|
key_validator = QRegularExpressionValidator(QRegularExpression(r"[A-Za-z.]+"))
|
|
self.key_input.setValidator(key_validator)
|
|
layout.addWidget(self.key_input)
|
|
|
|
self.type_combo = QComboBox()
|
|
self.type_combo.addItems(["int", "str", "float", "u32", "i32"])
|
|
layout.addWidget(self.type_combo)
|
|
|
|
self.value_input = QLineEdit()
|
|
self.value_input.setPlaceholderText("Value")
|
|
layout.addWidget(self.value_input)
|
|
|
|
delete_button = QPushButton("✕")
|
|
delete_button.setFixedSize(30, 30)
|
|
delete_button.clicked.connect(self.delete_clicked)
|
|
layout.addWidget(delete_button)
|
|
|
|
# Connect type change to validator update
|
|
self.type_combo.currentTextChanged.connect(self.update_validator)
|
|
|
|
# Initialize validator
|
|
self.update_validator(self.type_combo.currentText())
|
|
|
|
def delete_clicked(self) -> None:
|
|
self.deleted.emit(self)
|
|
|
|
def get_override_string(
|
|
self,
|
|
model_name=None,
|
|
quant_type=None,
|
|
output_path=None,
|
|
quantization_parameters=None,
|
|
) -> str: # Add arguments
|
|
key = self.key_input.text()
|
|
type_ = self.type_combo.currentText()
|
|
value = self.value_input.text()
|
|
|
|
dynamic_params = {
|
|
"{system.time.milliseconds}": lambda: str(int(time.time() * 1000)),
|
|
"{system.time.seconds}": lambda: str(int(time.time())),
|
|
"{system.date.iso}": lambda: datetime.now().strftime("%Y-%m-%d"),
|
|
"{system.datetime.iso}": lambda: datetime.now().isoformat(),
|
|
"{system.username}": lambda: os.getlogin(),
|
|
"{system.hostname}": lambda: socket.gethostname(),
|
|
"{system.platform}": lambda: platform.system(),
|
|
"{system.python.version}": lambda: platform.python_version(),
|
|
"{system.timezone}": lambda: time.tzname[time.daylight],
|
|
"{system.cpus}": lambda: str(os.cpu_count()),
|
|
"{system.memory.total}": lambda: str(psutil.virtual_memory().total),
|
|
"{system.memory.free}": lambda: str(psutil.virtual_memory().free),
|
|
"{system.filesystem.used}": lambda: str(shutil.disk_usage("/").used),
|
|
"{system.kernel.version}": lambda: platform.release(),
|
|
"{system.locale}": lambda: locale.getdefaultlocale()[0],
|
|
"{process.nice}": lambda: str(os.nice(0)),
|
|
"{model.name}": lambda: (
|
|
model_name if model_name is not None else "Unknown Model"
|
|
),
|
|
"{quant.type}": lambda: (
|
|
quant_type if quant_type is not None else "Unknown Quant"
|
|
),
|
|
"{output.path}": lambda: (
|
|
output_path if output_path is not None else "Unknown Output Path"
|
|
),
|
|
"{quant.kv}": lambda: (
|
|
quantization_parameters[0]
|
|
if quantization_parameters is not None
|
|
else False
|
|
),
|
|
"{quant.requantized}": lambda: (
|
|
quantization_parameters[1]
|
|
if quantization_parameters is not None
|
|
else False
|
|
),
|
|
"{quant.leave_output_tensor}": lambda: (
|
|
quantization_parameters[2]
|
|
if quantization_parameters is not None
|
|
else False
|
|
),
|
|
}
|
|
|
|
for param, func in dynamic_params.items():
|
|
value = value.replace(param, func())
|
|
|
|
return f"{key}={type_}:{value}"
|
|
|
|
def get_raw_override_string(self) -> str:
|
|
# Return the raw override string with placeholders intact
|
|
return f"{self.key_input.text()}={self.type_combo.currentText()}:{self.value_input.text()}"
|
|
|
|
def update_validator(self, type_) -> None:
|
|
if type_ == "int":
|
|
self.value_input.setValidator(QIntValidator())
|
|
elif type_ == "float":
|
|
self.value_input.setValidator(QDoubleValidator())
|
|
else: # str
|
|
self.value_input.setValidator(None)
|