Pakobbix/StartUI-oobabooga-webui
1
0
Fork 0
mirror of https://github.com/Pakobbix/StartUI-oobabooga-webui synced 2025-11-09 18:39:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed nvidia_less save & load + deepspeed nvme offload save & load
Some checks reported warnings
Package Release / package (ubuntu-latest) (push) Has been cancelled
Details
Package Release / package (windows-latest) (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Pakobbix 2023-05-17 01:20:17 +02:00
parent 2d6298c59b
commit 2e0acc1b3d
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
StartUI.py
View File

@ -521,8 +521,8 @@ class MainWindow(QMainWindow):
layout.addWidget(self.deepspeed_nvme_label, 33 + len(gpu_stats), 0) layout.addWidget(self.deepspeed_nvme_label, 33 + len(gpu_stats), 0)
# NVMe Current Offload Directory # NVMe Current Offload Directory
self.offload_directory = "none" self.selected_offload_directory = "none"
self.deepspeed_nvme_current_label = QLabel(f"Current Directory: {self.offload_directory}") self.deepspeed_nvme_current_label = QLabel(f"Current Directory: {self.selected_offload_directory}")
self.deepspeed_nvme_current_label.setVisible(False) self.deepspeed_nvme_current_label.setVisible(False)
self.deepspeed_nvme_current_label.setToolTip("The current NVMe offload directory.") self.deepspeed_nvme_current_label.setToolTip("The current NVMe offload directory.")
layout.addWidget(self.deepspeed_nvme_current_label, 33 + len(gpu_stats), 1) layout.addWidget(self.deepspeed_nvme_current_label, 33 + len(gpu_stats), 1)
@ -1076,6 +1076,8 @@ class MainWindow(QMainWindow):
if folder: if folder:
self.selected_offload_directory = folder self.selected_offload_directory = folder
self.deepspeed_nvme_current_label.setText(f"Current Directory Folder: {self.selected_offload_directory}") self.deepspeed_nvme_current_label.setText(f"Current Directory Folder: {self.selected_offload_directory}")
else:
self.selected_offload_directory = none
# def on_deepspeed_nvme_button_clicked(self): # def on_deepspeed_nvme_button_clicked(self):
# folder = QFileDialog.getExistingDirectory(self, "Offload Directory") # folder = QFileDialog.getExistingDirectory(self, "Offload Directory")
@ -1403,8 +1405,7 @@ class MainWindow(QMainWindow):
} }
if nvidia_gpu: if nvidia_gpu:
settings["gpu_vram"] = [slider.value() for slider in self.gpu_vram_sliders], # Saves the VRAM Values settings["gpu_vram"] = [slider.value() for slider in self.gpu_vram_sliders]
# Get the text entered in the text field # Get the text entered in the text field
profile_name = self.profile_name_textfield.text() profile_name = self.profile_name_textfield.text()
@ -1740,11 +1741,13 @@ class MainWindow(QMainWindow):
self.pre_layer_slider.setValue(int(settings.get("prelayer", 0))) self.pre_layer_slider.setValue(int(settings.get("prelayer", 0)))
self.use_autolaunch_checkbox.setChecked(settings.get("autolaunch", False)) self.use_autolaunch_checkbox.setChecked(settings.get("autolaunch", False))
self.use_network_checkbox.setChecked(settings.get("listen", False)) self.use_network_checkbox.setChecked(settings.get("listen", False))
if nvidia_gpu: if nvidia_gpu:
gpu_vram_settings = settings.get("gpu_vram", []) gpu_vram_settings = settings.get("gpu_vram", [])
for idx, slider in enumerate(self.gpu_vram_sliders): for idx, slider in enumerate(self.gpu_vram_sliders):
if idx < len(gpu_vram_settings): if idx < len(gpu_vram_settings):
slider.setValue(gpu_vram_settings[idx]) slider.setValue(gpu_vram_settings[idx])
self.use_extensions_checkbox.setChecked(settings.get("use_extension", False)) self.use_extensions_checkbox.setChecked(settings.get("use_extension", False))
extensions_settings = settings.get("extensions", []) extensions_settings = settings.get("extensions", [])
for i in range(self.extensions_list.count()): for i in range(self.extensions_list.count()):