fix: replace regexp to support unicode chars (#1424)

lllyasviel · Feb 10, 2024 · c32b9bd · c32b9bd
1 parent 98ba1d5
commit c32b9bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/launch_util.py b/modules/launch_util.py
@@ -15,7 +15,7 @@
 logging.getLogger("torch.distributed.nn").setLevel(logging.ERROR)  # sshh...
 logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
 
-re_requirement = re.compile(r"\s*([-_a-zA-Z0-9]+)\s*(?:==\s*([-+_.a-zA-Z0-9]+))?\s*")
+re_requirement = re.compile(r"\s*([-\w]+)\s*(?:==\s*([-+.\w]+))?\s*")
 
 python = sys.executable
 default_command_live = (os.environ.get('LAUNCH_LIVE_OUTPUT') == "1")