Spaces:
Running
on
Zero
Running
on
Zero
wangshuai6
commited on
Commit
·
79ac8fc
1
Parent(s):
9ef9890
init
Browse files
src/models/conditioner/qwen3_text_encoder.py
CHANGED
|
@@ -11,7 +11,7 @@ class Qwen3TextEncoder(BaseConditioner):
|
|
| 11 |
self.tokenizer = Qwen2Tokenizer.from_pretrained(weight_path, max_length=max_length, padding_side="right")
|
| 12 |
# self.model = Qwen3Model.from_pretrained(weight_path, attn_implementation="flex_attention").to(torch.bfloat16)
|
| 13 |
self.model = Qwen3Model.from_pretrained(weight_path).to(torch.bfloat16)
|
| 14 |
-
self.model.compile()
|
| 15 |
self.uncondition_embedding = None
|
| 16 |
self.embed_dim = embed_dim
|
| 17 |
self.max_length = max_length
|
|
|
|
| 11 |
self.tokenizer = Qwen2Tokenizer.from_pretrained(weight_path, max_length=max_length, padding_side="right")
|
| 12 |
# self.model = Qwen3Model.from_pretrained(weight_path, attn_implementation="flex_attention").to(torch.bfloat16)
|
| 13 |
self.model = Qwen3Model.from_pretrained(weight_path).to(torch.bfloat16)
|
| 14 |
+
# self.model.compile()
|
| 15 |
self.uncondition_embedding = None
|
| 16 |
self.embed_dim = embed_dim
|
| 17 |
self.max_length = max_length
|