Skip to content

Bug: device python object written in the config of the model card and keys sorted  #22

Description

@bergr7

🐛 Bug report

Summary

Python object is written in the model card's config file

Actual Behavior

device python object is written in the model card.

Also, keys are being sorted instead of keeping the original order.

See https://huggingface.co/flow-ai-llm/biomistral_slerp_7b

Expected behavior

python object not written in config of model card.
Keys ordered as in the original config.

To Reproduce

Merge configuration file (if relevant):

method: slerp
method_global_parameters:
  t: 0.5
base_model: BioMistral/BioMistral-Safetensors
models:
  - model: BioMistral/BioMistral-Safetensors
  - model: mistralai/Mistral-7B-Instruct-v0.2
tokenizer:
  mode: base
  interpolation_method: linear
directory_settings:
  output_dir: ./biomistral/biomistral_slerp_7b/
hf_token:
  token: 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
  trust_remote_code: True
device: cpu

Environment

  • OS: Ubuntu 22.04.4 LTS
  • Python version:Python 3.10.13
  • Library version: 0.1.0
  • Other relevant dependencies: NA

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

Status
In progress

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions