From 14713eb95ec7ba6db89cabdfff431b62089ea377 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Sun, 10 Sep 2023 22:08:55 -0700
Subject: [PATCH] fix oom

---
 fooocus_version.py   | 2 +-
 modules/expansion.py | 8 ++++++++
 update_log.md        | 4 ++++
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/fooocus_version.py b/fooocus_version.py
index d42c5beb..0ea49816 100644
--- a/fooocus_version.py
+++ b/fooocus_version.py
@@ -1 +1 @@
-version = '1.0.63'
+version = '1.0.64'
diff --git a/modules/expansion.py b/modules/expansion.py
index cc71c2ab..e5c16992 100644
--- a/modules/expansion.py
+++ b/modules/expansion.py
@@ -9,6 +9,7 @@ fooocus_magic_split = [
     ', intricate',
     '. The',
 ]
+dangrous_patterns = '[]【】()（）|:：'
 
 
 def safe_str(x):
@@ -18,6 +19,12 @@ def safe_str(x):
     return x.rstrip(",. \r\n")
 
 
+def remove_pattern(x, pattern):
+    for p in pattern:
+        x = x.replace(p, '')
+    return x
+
+
 class FooocusExpansion:
     def __init__(self):
         self.tokenizer = AutoTokenizer.from_pretrained(fooocus_expansion_path)
@@ -37,4 +44,5 @@ class FooocusExpansion:
         response = self.pipe(prompt, max_length=len(prompt) + 256)
         result = response[0]['generated_text'][len(origin):]
         result = safe_str(result)
+        result = remove_pattern(result, dangrous_patterns)
         return result
diff --git a/update_log.md b/update_log.md
index ce5bbd4b..9ea8f0e8 100644
--- a/update_log.md
+++ b/update_log.md
@@ -1,3 +1,7 @@
+### 1.0.64
+
+* Fixed a small OOM problem.
+
 ### 1.0.62
 
 * Change prompt expansion to suffix mode for better balance of semantic and style (and debugging).