From b403cb7300f288360023c2a1978b86a361934e24 Mon Sep 17 00:00:00 2001
From: hako-mikan <122196982+hako-mikan@users.noreply.github.com>
Date: Thu, 19 Jun 2025 23:05:20 +0900
Subject: [PATCH] #186

---
 scripts/lora_block_weight.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/scripts/lora_block_weight.py b/scripts/lora_block_weight.py
index 8a5756e..76b22a9 100644
--- a/scripts/lora_block_weight.py
+++ b/scripts/lora_block_weight.py
@@ -16,7 +16,7 @@ import modules.ui
 import modules.scripts as scripts
 from PIL import Image, ImageFont, ImageDraw
 import modules.shared as shared
-from modules import sd_models, images,cmd_args, extra_networks, devices
+from modules import sd_models, images,cmd_args, extra_networks, devices, launch_utils
 from modules.shared import cmd_opts, opts, state
 from modules.processing import process_images, Processed
 from modules.script_callbacks import CFGDenoiserParams, on_cfg_denoiser, on_ui_settings
@@ -45,9 +45,8 @@ xyelem = ""
 princ = False
 
 try:
-    from modules.ui import versions_html
-    forge = "forge" in versions_html()
-    reforge = "reForge" in versions_html()
+    forge = launch_utils.git_tag()[0:2] == "f2"
+    reforge = launch_utils.git_tag()[0:2] == "f1" or launch_utils.git_tag() == "classic"
 except:
     forge = reforge = False