Merge pull request #1260 from bghira/feature/sana-prompt-cleaning

add beautifulsoup4 for prompt cleaning in Sana
bghira · Jan 2, 2025 · 13c405f · 13c405f
2 parents 2ea9b57 + 7684851
commit 13c405f
Show file tree

Hide file tree

Showing 6 changed files with 104 additions and 5 deletions.
diff --git a/install/apple/poetry.lock b/install/apple/poetry.lock
diff --git a/install/apple/pyproject.toml b/install/apple/pyproject.toml
@@ -46,6 +46,7 @@ sentencepiece = "^0.2.0"
 torchao = "^0.7.0"
 torchaudio = "^2.5.0"
 atomicwrites = "^1.4.1"
+beautifulsoup4 = "^4.12.3"
 
 
 [build-system]

diff --git a/install/rocm/poetry.lock b/install/rocm/poetry.lock
diff --git a/install/rocm/pyproject.toml b/install/rocm/pyproject.toml
@@ -46,6 +46,7 @@ fastapi = {extras = ["standard"], version = "^0.115.0"}
 bitsandbytes = "^0.44.1"
 atomicwrites = "^1.4.1"
 torchao = "^0.7.0"
+beautifulsoup4 = "^4.12.3"
 
 
 [build-system]

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -51,6 +51,7 @@ lm-eval = "^0.4.4"
 nvidia-cudnn-cu12 = "*"
 nvidia-nccl-cu12 = "*"
 atomicwrites = "^1.4.1"
+beautifulsoup4 = "^4.12.3"