Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -41,6 +41,7 @@ def get_lora_sd_pipeline(
|
|
41 |
def split_prompt(prompt, tokenizer, max_length=77):
|
42 |
print(prompt)
|
43 |
print(type(prompt))
|
|
|
44 |
tokens = tokenizer(prompt, truncation=False)["input_ids"]
|
45 |
chunks = [tokens[i:i + max_length] for i in range(0, len(tokens), max_length)]
|
46 |
return chunks
|
|
|
41 |
def split_prompt(prompt, tokenizer, max_length=77):
|
42 |
print(prompt)
|
43 |
print(type(prompt))
|
44 |
+
print(prompt.value)
|
45 |
tokens = tokenizer(prompt, truncation=False)["input_ids"]
|
46 |
chunks = [tokens[i:i + max_length] for i in range(0, len(tokens), max_length)]
|
47 |
return chunks
|