Add llms subdir + update README (#145)

* add llms subdir + update README

* nits

* use same pre-commit as mlx

* update readmes a bit

* format
This commit is contained in:
Awni Hannun
2023-12-20 10:22:25 -08:00
committed by GitHub
parent aed14618ca
commit 27c0a8c002
62 changed files with 164 additions and 146 deletions

View File

@@ -1,8 +1,9 @@
import argparse
from transformers import AutoModelForCausalLM
import json
import numpy as np
import torch
import json
from transformers import AutoModelForCausalLM
def replace_key(key: str) -> str:

View File

@@ -1,6 +1,7 @@
import argparse
from dataclasses import dataclass
import json
from dataclasses import dataclass
import mlx.core as mx
import mlx.nn as nn
from mlx.utils import tree_unflatten