extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-26 12:52:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix embedding extraction for weights, commas, etc. This fixes the recent change where 'world' would match 'rld'.

Browse Source
This commit is contained in:
cmdr2 2023-08-29 10:23:05 +05:30
parent a7b0858b22
commit b1cb03962c
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ui/media/js/main.js
View File

@ -1312,9 +1312,11 @@ function getCurrentUserRequest() {
} }
function setEmbeddings(task) { function setEmbeddings(task) {
let prompt = task.reqBody.prompt.toLowerCase() let prompt = task.reqBody.prompt
let negativePrompt = task.reqBody.negative_prompt.toLowerCase() let negativePrompt = task.reqBody.negative_prompt
let overallPrompt = prompt + " " + negativePrompt let overallPrompt = (prompt + " " + negativePrompt).toLowerCase()
overallPrompt = overallPrompt.replaceAll(/[^a-z0-9\.]/g, " ") // only allow alpha-numeric and dots
overallPrompt = overallPrompt.split(" ")
let embeddingsTree = modelsOptions["embeddings"] let embeddingsTree = modelsOptions["embeddings"]
let embeddings = [] let embeddings = []