Merge remote-tracking branch 'refs/remotes/origin/dev' into dev

This commit is contained in:
oobabooga 2024-01-26 06:12:19 -08:00
commit de387069da
10 changed files with 10 additions and 10 deletions

View File

@ -162,7 +162,7 @@ def huggingface_loader(model_name):
# DeepSpeed ZeRO-3 # DeepSpeed ZeRO-3
elif shared.args.deepspeed: elif shared.args.deepspeed:
model = LoaderClass.from_pretrained(path_to_model, torch_dtype=params['torch_dtype']) model = LoaderClass.from_pretrained(path_to_model, torch_dtype=params['torch_dtype'], trust_remote_code=params['trust_remote_code'])
model = deepspeed.initialize(model=model, config_params=ds_config, model_parameters=None, optimizer=None, lr_scheduler=None)[0] model = deepspeed.initialize(model=model, config_params=ds_config, model_parameters=None, optimizer=None, lr_scheduler=None)[0]
model.module.eval() # Inference model.module.eval() # Inference
logger.info(f'DeepSpeed ZeRO-3 is enabled: {is_deepspeed_zero3_enabled()}') logger.info(f'DeepSpeed ZeRO-3 is enabled: {is_deepspeed_zero3_enabled()}')

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12; platform_system != "Darwin" and platform_machine != "x86_64" exllamav2==0.0.12; platform_system != "Darwin" and platform_machine != "x86_64"
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12; platform_system == "Windows" or python_version < "3.10" or python_version > "3.11" or platform_machine != "x86_64" exllamav2==0.0.12; platform_system == "Windows" or python_version < "3.10" or python_version > "3.11" or platform_machine != "x86_64"
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12; platform_system == "Windows" or python_version < "3.10" or python_version > "3.11" or platform_machine != "x86_64" exllamav2==0.0.12; platform_system == "Windows" or python_version < "3.10" or python_version > "3.11" or platform_machine != "x86_64"
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12 exllamav2==0.0.12
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12 exllamav2==0.0.12
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12 exllamav2==0.0.12
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12 exllamav2==0.0.12
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12; platform_system != "Darwin" and platform_machine != "x86_64" exllamav2==0.0.12; platform_system != "Darwin" and platform_machine != "x86_64"
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown

View File

@ -4,7 +4,7 @@ datasets
einops einops
exllamav2==0.0.12 exllamav2==0.0.12
gradio==3.50.* gradio==3.50.*
hqq==0.1.2 hqq==0.1.2.post1
jinja2==3.1.2 jinja2==3.1.2
lm_eval==0.3.0 lm_eval==0.3.0
markdown markdown