Spaces:

AbstractPhil
/

shunt-adapter-testing

Running on Zero

AbstractPhil commited on 8 days ago

Commit

f547ef2

1 Parent(s): 9cb35be

yes

Files changed (2) hide show

__pycache__/two_stream_shunt_adapter.cpython-310.pyc CHANGED Viewed

Binary files a/__pycache__/two_stream_shunt_adapter.cpython-310.pyc and b/__pycache__/two_stream_shunt_adapter.cpython-310.pyc differ

app.py CHANGED Viewed

@@ -143,7 +143,13 @@ def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noi
         pipe.scheduler = SCHEDULERS[scheduler_name].from_config(pipe.scheduler.config)
     # Get T5 embeddings for semantic understanding
-    t5_ids = t5_tok(prompt, return_tensors="pt", padding=True, truncation=True).input_ids.to(device)
     t5_seq = t5_mod(t5_ids).last_hidden_state
     # Get proper SDXL CLIP embeddings

         pipe.scheduler = SCHEDULERS[scheduler_name].from_config(pipe.scheduler.config)
     # Get T5 embeddings for semantic understanding
+    t5_ids = t5_tok(
+        prompt,
+        return_tensors="pt",
+        padding="max_length",
+        max_length=77,  # Match CLIP's standard length
+        truncation=True
+    ).input_ids.to(device)
     t5_seq = t5_mod(t5_ids).last_hidden_state
     # Get proper SDXL CLIP embeddings