Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "2eab1132062caecf8954d351cc2c45d4bec4a77f"
Unverified Commit 4fcd0bc7 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[chore] remove extra validation check in determine_device_map (#12176) 

remove extra validation check in determine_device_map
parent 7993be9e
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
src/diffusers/models/model_loading_utils.py
View file @ 4fcd0bc7
...@@ -112,9 +112,6 @@ def _determine_device_map( ...@@ -112,9 +112,6 @@ def _determine_device_map(
device_map_kwargs["max_memory"] = max_memory device_map_kwargs["max_memory"] = max_memory
device_map = infer_auto_device_map(model, dtype=target_dtype, **device_map_kwargs) device_map = infer_auto_device_map(model, dtype=target_dtype, **device_map_kwargs)
if hf_quantizer is not None:
hf_quantizer.validate_environment(device_map=device_map)
return device_map return device_map
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment