Commit
•
5d2d710
1
Parent(s):
ee22f80
修复compel名称 (#2)
Browse files- 修复compel名称 (0ddba2f1d14f7fee6b0bb9b6beeb3b727f37ef94)
Co-authored-by: animelover <[email protected]>
app.py
CHANGED
@@ -257,7 +257,7 @@ def infer(
|
|
257 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
258 |
generator = torch.Generator().manual_seed(seed)
|
259 |
# 初始化 Compel 实例
|
260 |
-
|
261 |
tokenizer=[pipe.tokenizer, pipe.tokenizer_2],
|
262 |
text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
|
263 |
returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED,
|
|
|
257 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
258 |
generator = torch.Generator().manual_seed(seed)
|
259 |
# 初始化 Compel 实例
|
260 |
+
compel = Compel(
|
261 |
tokenizer=[pipe.tokenizer, pipe.tokenizer_2],
|
262 |
text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
|
263 |
returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED,
|