fixes

vms/config.py

@@ -507,7 +507,7 @@ class TrainingConfig:
         args.extend(["--train_steps", str(self.train_steps)])
         
         # LoRA specific arguments
-        ifself.training_type == "lora":
+        if self.training_type == "lora":
             args.extend(["--rank", str(self.lora_rank)])
             args.extend(["--lora_alpha", str(self.lora_alpha)])
             args.extend(["--target_modules"] + self.target_modules)

vms/tabs/__init__.py

@@ -7,6 +7,7 @@ from .split_tab import SplitTab
 from .caption_tab import CaptionTab
 from .train_tab import TrainTab
 from .monitor_tab import MonitorTab
+from .preview_tab import PreviewTab
 from .manage_tab import ManageTab
 
 __all__ = [
@@ -15,5 +16,6 @@ __all__ = [
     'CaptionTab',
     'TrainTab',
     'MonitorTab',
+    'PreviewTab',
     'ManageTab'
 ]

vms/ui/video_trainer_ui.py

@@ -28,7 +28,7 @@ from ..utils import (
     format_media_title,
     TrainingLogParser
 )
-from ..tabs import ImportTab, SplitTab, CaptionTab, TrainTab, MonitorTab, ManageTab
+from ..tabs import ImportTab, SplitTab, CaptionTab, TrainTab, MonitorTab, PreviewTab, ManageTab
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
@@ -113,6 +113,7 @@ class VideoTrainerUI:
                 self.tabs["caption_tab"] = CaptionTab(self)
                 self.tabs["train_tab"] = TrainTab(self)
                 self.tabs["monitor_tab"] = MonitorTab(self)
+                self.tabs["preview_tab"] = PreviewTab(self)
                 self.tabs["manage_tab"] = ManageTab(self)
                 
                 # Create tab UI components