mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-22 16:17:57 +01:00
Merge remote-tracking branch 'refs/remotes/origin/dev' into dev
This commit is contained in:
commit
de387069da
@ -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()}')
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user