Skip to content

Commit e2194b3

Browse files
committed
fix conflict
2 parents 3bf4272 + dcfcc2a commit e2194b3

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

optimum/exporters/openvino/stateful.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,7 @@
2222
from packaging import version
2323
import openvino as ov
2424
from openvino.runtime import opset13
25-
<<<<<<< HEAD
2625
from optimum.intel.utils.import_utils import _openvino_version, is_openvino_version
27-
=======
28-
from optimum.intel.utils.import_utils import is_openvino_version
29-
>>>>>>> Fix for bloom family
3026
from optimum.utils.normalized_config import NormalizedConfigManager
3127

3228

@@ -228,14 +224,8 @@ def patch_stateful(config: PretrainedConfig, ov_model: ov.Model):
228224

229225
fuse_cache_reorder(ov_model, not_kv_inputs, key_value_input_names, batch_dim)
230226

231-
<<<<<<< HEAD
232227
normalized_config = NormalizedConfigManager.get_normalized_config_class(config.model_type)(config)
233228
num_attention_heads = normalized_config.num_attention_heads if config.model_type == "bloom" else 1
234-
=======
235-
normalized_config = NormalizedConfigManager.get_normalized_config_class(model.config.model_type)(model.config)
236-
num_attention_heads = normalized_config.num_attention_heads if model.config.model_type == 'bloom' else 1
237-
>>>>>>> Fix for bloom family
238-
239229
make_stateful(
240230
ov_model, not_kv_inputs, key_value_input_names, key_value_output_names, batch_dim, num_attention_heads, None
241231
)

0 commit comments

Comments
 (0)