Spaces:

adamelliotfields
/

diffusion

Running on Zero

adamelliotfields commited on Sep 17

Commit

1a7f234

•

1 Parent(s): f70898c

Don't truncate prompts

Files changed (1) hide show

lib/inference.py CHANGED Viewed

@@ -235,6 +235,7 @@ def generate(
     compel = Compel(
         device=pipe.device,
         tokenizer=pipe.tokenizer,
         text_encoder=pipe.text_encoder,
         returned_embeddings_type=EMBEDDINGS_TYPE,
         dtype_for_device_getter=lambda _: pipe.dtype,

     compel = Compel(
         device=pipe.device,
         tokenizer=pipe.tokenizer,
+        truncate_long_prompts=False,
         text_encoder=pipe.text_encoder,
         returned_embeddings_type=EMBEDDINGS_TYPE,
         dtype_for_device_getter=lambda _: pipe.dtype,