wangshuai6 commited on
Commit
79ac8fc
·
1 Parent(s): 9ef9890
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