Skip to content

Commit e0cbfab

Browse files
authored
Merge pull request #606 from eltociear/patch-3
Update modeling_lm.py
2 parents ae612e3 + fcf001c commit e0cbfab

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

FlagEmbedding/llm_embedder/src/lm/modeling_lm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(self, model_name_or_path=None, padding_side="left", dtype="bf16", c
3838
except ValueError:
3939
self.model = AutoModelForSeq2SeqLM.from_pretrained(model_name_or_path, cache_dir=cache_dir, torch_dtype=dtype, trust_remote_code=True, device_map=device_map)
4040

41-
# if device_map is specified, we don't need to move the model to any specifc gpu
41+
# if device_map is specified, we don't need to move the model to any specific gpu
4242
if device_map is None:
4343
if accelerator is not None:
4444
device = accelerator.device

0 commit comments

Comments
 (0)