Skip to content

Commit

Permalink
fixes memory preferences
Browse files Browse the repository at this point in the history
  • Loading branch information
@w4ffl35
w4ffl35 committed Oct 18, 2023
1 parent da928b2 commit 6af250a
Show file tree
Hide file tree
Showing 4 changed files with 303 additions and 67 deletions.
70 changes: 70 additions & 0 deletions src/airunner/widgets/memory_preferences/memory_preferences_widget.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,70 @@
from airunner.widgets.base_widget import BaseWidget
from airunner.widgets.memory_preferences.templates.memory_preferences_ui import Ui_memory_preferences


class MemoryPreferencesWidget(BaseWidget):
widget_class_ = Ui_memory_preferences

def __init__(self, **kwargs):
super().__init__(**kwargs)

self.ui.use_accelerated_transformers.blockSignals(True)
self.ui.use_attention_slicing.blockSignals(True)
self.ui.use_enable_sequential_cpu_offload.blockSignals(True)
self.ui.enable_model_cpu_offload.blockSignals(True)
self.ui.use_lastchannels.blockSignals(True)
self.ui.use_tf32.blockSignals(True)
self.ui.use_tiled_vae.blockSignals(True)
self.ui.use_enable_vae_slicing.blockSignals(True)

self.ui.use_accelerated_transformers.setChecked(self.settings_manager.memory_settings.use_accelerated_transformers is True)
self.ui.use_attention_slicing.setChecked(self.settings_manager.memory_settings.use_attention_slicing is True)
self.ui.use_enable_sequential_cpu_offload.setChecked(
self.settings_manager.memory_settings.use_enable_sequential_cpu_offload is True)
self.ui.enable_model_cpu_offload.setChecked(
self.settings_manager.memory_settings.enable_model_cpu_offload is True
)
self.ui.use_lastchannels.setChecked(self.settings_manager.memory_settings.use_last_channels is True)
self.ui.use_tf32.setChecked(self.settings_manager.memory_settings.use_tf32 is True)
self.ui.use_tiled_vae.setChecked(self.settings_manager.memory_settings.use_tiled_vae is True)
self.ui.use_enable_vae_slicing.setChecked(self.settings_manager.memory_settings.use_enable_vae_slicing is True)

self.ui.use_accelerated_transformers.blockSignals(False)
self.ui.use_attention_slicing.blockSignals(False)
self.ui.use_enable_sequential_cpu_offload.blockSignals(False)
self.ui.enable_model_cpu_offload.blockSignals(False)
self.ui.use_lastchannels.blockSignals(False)
self.ui.use_tf32.blockSignals(False)
self.ui.use_tiled_vae.blockSignals(False)
self.ui.use_enable_vae_slicing.blockSignals(False)

def action_toggled_tile_vae(self, val):
self.settings_manager.set_value("memory_settings.use_tiled_vae", val)

def action_toggled_tf32(self, val):
self.settings_manager.set_value("memory_settings.use_tf32", val)

def action_toggled_last_memory(self, val):
self.settings_manager.set_value("memory_settings.use_last_channels", val)

def action_toggled_vae_slicing(self, val):
self.settings_manager.set_value("memory_settings.use_enable_vae_slicing", val)

def action_toggled_sequential_cpu_offload(self, val):
self.settings_manager.set_value("memory_settings.use_enable_sequential_cpu_offload", val)

def action_toggled_attention_slicing(self, val):
self.settings_manager.set_value("memory_settings.use_attention_slicing", val)

def action_toggled_accelerated_transformers(self, val):
self.settings_manager.set_value("memory_settings.use_accelerated_transformers", val)

def action_button_clicked_optimize_memory_settings(self):
self.ui.use_accelerated_transformers.setChecked(True)
self.ui.use_attention_slicing.setChecked(False)
self.ui.use_lastchannels.setChecked(True)
self.ui.use_enable_sequential_cpu_offload.setChecked(False)
self.ui.enable_model_cpu_offload.setChecked(False)
self.ui.use_tf32.setChecked(False)
self.ui.use_tiled_vae.setChecked(True)
self.ui.use_enable_vae_slicing.setChecked(True)
164 changes: 160 additions & 4 deletions src/airunner/widgets/memory_preferences/templates/memory_preferences.ui
View file
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
<class>Form</class>
<widget class="QWidget" name="Form">
<class>memory_preferences</class>
<widget class="QWidget" name="memory_preferences">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>352</width>
<height>504</height>
<height>544</height>
</rect>
</property>
<property name="windowTitle">
Expand Down Expand Up @@ -365,5 +365,161 @@
</layout>
</widget>
<resources/>
<connections/>
<connections>
<connection>
<sender>use_accelerated_transformers</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_accelerated_transformers(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>145</x>
<y>62</y>
</hint>
<hint type="destinationlabel">
<x>5</x>
<y>5</y>
</hint>
</hints>
</connection>
<connection>
<sender>use_attention_slicing</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_attention_slicing(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>86</x>
<y>142</y>
</hint>
<hint type="destinationlabel">
<x>2</x>
<y>39</y>
</hint>
</hints>
</connection>
<connection>
<sender>use_lastchannels</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_last_memory(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>163</x>
<y>198</y>
</hint>
<hint type="destinationlabel">
<x>271</x>
<y>6</y>
</hint>
</hints>
</connection>
<connection>
<sender>use_enable_sequential_cpu_offload</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_sequential_cpu_offload(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>124</x>
<y>255</y>
</hint>
<hint type="destinationlabel">
<x>258</x>
<y>0</y>
</hint>
</hints>
</connection>
<connection>
<sender>enable_model_cpu_offload</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_sequential_cpu_offload(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>75</x>
<y>321</y>
</hint>
<hint type="destinationlabel">
<x>2</x>
<y>268</y>
</hint>
</hints>
</connection>
<connection>
<sender>use_tf32</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_tf32(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>37</x>
<y>382</y>
</hint>
<hint type="destinationlabel">
<x>-3</x>
<y>318</y>
</hint>
</hints>
</connection>
<connection>
<sender>use_enable_vae_slicing</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_vae_slicing(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>93</x>
<y>438</y>
</hint>
<hint type="destinationlabel">
<x>0</x>
<y>388</y>
</hint>
</hints>
</connection>
<connection>
<sender>use_tiled_vae</sender>
<signal>toggled(bool)</signal>
<receiver>memory_preferences</receiver>
<slot>action_toggled_tile_vae(bool)</slot>
<hints>
<hint type="sourcelabel">
<x>40</x>
<y>495</y>
</hint>
<hint type="destinationlabel">
<x>-1</x>
<y>440</y>
</hint>
</hints>
</connection>
<connection>
<sender>optimize_memory_button</sender>
<signal>clicked()</signal>
<receiver>memory_preferences</receiver>
<slot>action_button_clicked_optimize_memory_settings()</slot>
<hints>
<hint type="sourcelabel">
<x>155</x>
<y>28</y>
</hint>
<hint type="destinationlabel">
<x>176</x>
<y>-14</y>
</hint>
</hints>
</connection>
</connections>
<slots>
<slot>action_button_clicked_optimize_memory_settings()</slot>
<slot>action_toggled_accelerated_transformers(bool)</slot>
<slot>action_toggled_attention_slicing(bool)</slot>
<slot>action_toggled_last_memory(bool)</slot>
<slot>action_toggled_sequential_cpu_offload(bool)</slot>
<slot>action_toggled_model_cpu_offload(bool)</slot>
<slot>action_toggled_tf32(bool)</slot>
<slot>action_toggled_vae_slicing(bool)</slot>
<slot>action_toggled_tile_vae(bool)</slot>
</slots>
</ui>
Loading

0 comments on commit 6af250a

Please sign in to comment.