Skip to content

Commit b2983b4

Browse files
jianan-gusywangyi
andauthored
fix stable diffusers textual_inversion finetune accurary issue. (#1533) (#1594)
the generated imaged by the finetune model unrelated with prompt Signed-off-by: Wang, Yi A <[email protected]> Co-authored-by: Wang, Yi <[email protected]>
1 parent ee7dc34 commit b2983b4

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

intel_extension_for_pytorch/nn/utils/_weight_cast.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import torch
22
import torch.nn as nn
3+
import sys
34
from intel_extension_for_pytorch.optim import _optimizer_utils, _lamb
45
import types
56
from ._model_convert import _LSTM
@@ -112,11 +113,16 @@ def convert(m):
112113
for name, para in sub_m.named_parameters():
113114
cast_attr(sub_m, name, master_weight_split, params_attr, optimizer)
114115
return m
115-
116+
117+
def isCLIPTextEmbeddings(m):
118+
mod = 'transformers.models.clip.modeling_clip'
119+
return mod in sys.modules and hasattr(sys.modules[mod], 'CLIPTextEmbeddings') and isinstance(m, sys.modules[mod].CLIPTextEmbeddings)
120+
116121
def convert_rec(m):
117122
new_m = convert(m)
118123
for name, sub_m in m.named_children():
119-
setattr(new_m, name, convert_rec(sub_m))
124+
if not isCLIPTextEmbeddings(sub_m):
125+
setattr(new_m, name, convert_rec(sub_m))
120126
return new_m
121127

122128
casted_model, casted_optimizer, params_attr = convert_rec(module), optimizer, params_attr

0 commit comments

Comments
 (0)