Fixed error with wan models when caching text embeddings
This commit is contained in:
1 parent
5c37db04f9
commit
115f0a3670
1 file changed
+2
-2
@@ -654,10 +654,10 @@ class Wan21(BaseModel):
|
||||
return latents.to(device, dtype=dtype)
|
||||
|
||||
def get_model_has_grad(self):
|
||||
return self.model.proj_out.weight.requires_grad
|
||||
return False
|
||||
|
||||
def get_te_has_grad(self):
|
||||
return self.text_encoder.encoder.block[0].layer[0].SelfAttention.q.weight.requires_grad
|
||||
return False
|
||||
|
||||
def save_model(self, output_path, meta, save_dtype):
|
||||
# only save the unet
|
||||
|
||||
Reference in new issue
Block a user