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
|