dezzman commited on
Commit
ad3015c
·
verified ·
1 Parent(s): 0211b29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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