dezzman commited on
Commit
b9af0ed
·
verified ·
1 Parent(s): d52ef02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -39,6 +39,9 @@ def get_lora_sd_pipeline(
39
  return pipe
40
 
41
  def split_prompt(prompt, tokenizer, max_length=77):
 
 
 
42
  tokens = tokenizer(prompt, truncation=False)["input_ids"]
43
  chunks = [tokens[i:i + max_length] for i in range(0, len(tokens), max_length)]
44
  return chunks
 
39
  return pipe
40
 
41
  def split_prompt(prompt, tokenizer, max_length=77):
42
+ print(prompt)
43
+ print(type(prompt))
44
+ print(str(prompt))
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