huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-09 11:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format

Browse Source
This commit is contained in:
regisss 2025-04-28 07:50:15 +00:00
parent 790a3b5ed2
commit 3e9f5ba159
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backends/gaudi/server/text_generation_server/models/causal_lm.py
View File

@ -882,7 +882,12 @@ class CausalLM(Model):
if load_to_meta: if load_to_meta:
# model loaded to meta is managed differently # model loaded to meta is managed differently
checkpoints_json = tempfile.NamedTemporaryFile(suffix=".json", mode="+w") checkpoints_json = tempfile.NamedTemporaryFile(suffix=".json", mode="+w")
checkpoint_files = [str(f) for f in weight_files(model_id, revision=revision, extension=".safetensors")] checkpoint_files = [
str(f)
for f in weight_files(
model_id, revision=revision, extension=".safetensors"
)
]
data = {"type": "ds_model", "checkpoints": checkpoint_files, "version": 1.0} data = {"type": "ds_model", "checkpoints": checkpoint_files, "version": 1.0}
json.dump(data, checkpoints_json) json.dump(data, checkpoints_json)
checkpoints_json.flush() checkpoints_json.flush()

7
backends/gaudi/server/text_generation_server/models/vlm_causal_lm.py
View File

@ -1017,7 +1017,12 @@ class VlmCausalLM(Model):
if load_to_meta: if load_to_meta:
# model loaded to meta is managed differently # model loaded to meta is managed differently
checkpoints_json = tempfile.NamedTemporaryFile(suffix=".json", mode="+w") checkpoints_json = tempfile.NamedTemporaryFile(suffix=".json", mode="+w")
checkpoint_files = [str(f) for f in weight_files(model_id, revision=revision, extension=".safetensors")] checkpoint_files = [
str(f)
for f in weight_files(
model_id, revision=revision, extension=".safetensors"
)
]
data = {"type": "ds_model", "checkpoints": checkpoint_files, "version": 1.0} data = {"type": "ds_model", "checkpoints": checkpoint_files, "version": 1.0}
json.dump(data, checkpoints_json) json.dump(data, checkpoints_json)
checkpoints_json.flush() checkpoints_json.flush()