{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "$id": "https://iinventedlight.com/library/artifacts/hf-from-pretrained.schema.json",
  "title": "Hugging Face from_pretrained common options (summary)",
  "type": "object",
  "additionalProperties": true,
  "properties": {
    "model_id_or_path": {
      "type": "string"
    },
    "device_map": {
      "type": [
        "string",
        "object",
        "null"
      ]
    },
    "torch_dtype": {
      "type": [
        "string",
        "null"
      ]
    },
    "low_cpu_mem_usage": {
      "type": [
        "boolean",
        "null"
      ]
    },
    "source_slug": {
      "type": "string"
    }
  },
  "required": [
    "model_id_or_path",
    "source_slug"
  ]
}