mlx-examples

zhangyiss/mlx-examples

Fork 0

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-31 03:19:23 +08:00

Commit Graph

Select branches

Hide Pull Requests

awq

awq-tq

dist-eval

distributed-layers

flux-dist-improv

flux-qlora

load-gguf

main

openlm

packed-quants

#1

#10

#1001

#1003

#1004

#1006

#1009

#1012

#1013

#1014

#1015

#1016

#102

#1020

#1023

#1024

#1026

#1027

#1028

#103

#1035

#1036

#1037

#1038

#1040

#1045

#1047

#1048

#1048

#1049

#105

#1050

#1054

#1055

#106

#1061

#1062

#1063

#1065

#1068

#1069

#107

#1070

#1072

#1074

#1075

#1078

#1079

#108

#1080

#1081

#1082

#1085

#1089

#1090

#1092

#1093

#1094

#1096

#1099

#110

#1100

#1103

#1105

#1113

#1114

#1115

#1117

#1118

#1119

#112

#1121

#1122

#1125

#1128

#1129

#113

#1131

#1132

#1133

#1135

#1137

#1140

#1141

#1143

#1144

#1145

#1146

#1148

#1149

#115

#1152

#1153

#1154

#1155

#1156

#1157

#1158

#1159

#116

#1163

#1164

#1166

#1173

#1174

#1176

#1178

#1180

#1189

#1191

#1192

#1193

#1194

#1196

#1197

#1199

#12

#1200

#1202

#1204

#1205

#1206

#1208

#1209

#121

#1210

#1211

#1212

#1213

#1214

#1215

#1216

#1217

#1220

#1222

#1225

#1228

#1229

#123

#1230

#1230

#1231

#1231

#1233

#1234

#1235

#1240

#1241

#1242

#1246

#1249

#1250

#1251

#1253

#1257

#1259

#1260

#1263

#1265

#1267

#1270

#1271

#1272

#1273

#1276

#1277

#1278

#1279

#1280

#1283

#1287

#129

#1291

#1294

#1295

#1297

#1298

#1299

#1300

#1301

#1302

#1304

#1305

#1308

#1311

#1312

#1314

#1315

#1316

#1319

#1321

#1322

#1323

#1324

#1325

#1326

#133

#1330

#1331

#1332

#1336

#1338

#1339

#134

#1340

#1342

#1346

#1348

#1349

#1350

#1353

#1358

#1359

#136

#1364

#1365

#1367

#1370

#1371

#1375

#1375

#1377

#1377

#1383

#1383

#1385

#1386

#1387

#14

#140

#141

#144

#145

#147

#149

#151

#152

#153

#154

#157

#158

#159

#160

#161

#162

#163

#164

#165

#166

#167

#169

#171

#172

#173

#174

#176

#177

#178

#180

#183

#184

#186

#187

#189

#19

#190

#191

#192

#193

#195

#197

#198

#2

#200

#201

#202

#203

#205

#208

#211

#213

#214

#215

#219

#22

#221

#222

#227

#229

#23

#231

#234

#235

#237

#238

#24

#240

#241

#242

#243

#245

#248

#250

#251

#252

#253

#254

#255

#257

#260

#263

#264

#265

#266

#269

#27

#270

#271

#272

#274

#275

#276

#278

#282

#284

#285

#287

#290

#291

#292

#293

#294

#295

#30

#300

#301

#302

#303

#306

#307

#308

#309

#310

#311

#312

#315

#318

#319

#32

#320

#321

#325

#326

#33

#331

#333

#335

#337

#338

#340

#342

#343

#347

#350

#351

#352

#353

#354

#358

#36

#360

#361

#363

#364

#365

#366

#369

#37

#373

#375

#377

#378

#379

#380

#386

#387

#388

#389

#391

#392

#393

#396

#397

#398

#399

#40

#401

#405

#408

#409

#41

#41

#411

#413

#414

#415

#417

#418

#419

#42

#420

#421

#424

#426

#427

#429

#43

#431

#432

#433

#439

#441

#443

#445

#446

#449

#45

#450

#451

#453

#455

#457

#458

#461

#462

#466

#467

#468

#469

#47

#470

#471

#472

#474

#475

#479

#48

#482

#483

#483

#486

#489

#491

#494

#495

#496

#497

#498

#5

#50

#501

#502

#503

#505

#506

#509

#51

#510

#514

#515

#516

#518

#519

#52

#520

#521

#522

#523

#528

#53

#530

#531

#534

#536

#539

#541

#544

#545

#546

#547

#548

#549

#55

#552

#555

#558

#562

#563

#565

#566

#569

#570

#571

#572

#573

#574

#577

#578

#580

#581

#585

#589

#590

#591

#592

#595

#596

#599

#602

#603

#604

#608

#609

#610

#611

#613

#614

#62

#621

#623

#628

#632

#633

#634

#636

#639

#64

#640

#643

#644

#645

#648

#650

#651

#654

#657

#66

#661

#665

#666

#667

#668

#670

#673

#674

#675

#676

#679

#680

#681

#682

#683

#684

#685

#687

#688

#69

#690

#691

#693

#694

#697

#698

#7

#701

#702

#703

#705

#707

#708

#711

#712

#715

#716

#717

#719

#720

#721

#729

#73

#731

#735

#736

#739

#74

#740

#743

#744

#746

#749

#75

#752

#753

#758

#76

#760

#763

#766

#77

#770

#773

#775

#778

#779

#78

#782

#789

#79

#790

#792

#793

#794

#797

#798

#8

#80

#800

#802

#803

#806

#807

#810

#813

#817

#818

#82

#821

#822

#824

#825

#827

#828

#830

#831

#833

#835

#837

#838

#839

#84

#840

#85

#851

#852

#853

#855

#856

#857

#86

#863

#867

#87

#871

#877

#879

#88

#882

#885

#886

#888

#889

#89

#890

#891

#895

#898

#899

#90

#902

#903

#904

#905

#906

#907

#911

#913

#914

#915

#915

#920

#923

#926

#93

#931

#932

#935

#936

#937

#94

#940

#942

#945

#946

#948

#949

#954

#955

#956

#957

#96

#960

#961

#962

#963

#965

#969

#97

#971

#971

#973

#979

#98

#981

#983

#984

#989

#99

#990

#991

#993

#995

#996

#998

c5cb83b9a0 Update lora readme.md, update mistral-7b-v0.1 to mistral-7b-v0.2-instruct madroid 2023-12-25 22:11:46 +0800
447abb2b0f QWEN: Fix unsupported ScalarType BFloat16 (#187) Yifan 2023-12-25 22:10:01 +0800
738448c2d4

QWEN: Fix unsupported ScalarType BFloat16 (#187) Yifan 2023-12-25 22:10:01 +0800
b9607f9510

QWEN: Fix unsupported ScalarType BFloat16 Yifan 2023-12-25 20:18:26 +0800
35b219b7ed

Merge branch 'ml-explore:main' into main LeonEricsson 2023-12-25 11:27:08 +0100
c190af6dc6 updated README (#184) Vidyasagar Bhargava 2023-12-24 19:49:53 +0530
647e48870a

updated README (#184) Vidyasagar Bhargava 2023-12-24 19:49:53 +0530
a43afd3234 updated README vidyasagarbhargava 2023-12-24 13:51:27 +0530
efa9837195 Mixtral: Stop at EOS token (#183) devonthomas35 2023-12-23 21:25:42 -0800
939086e6a3

Mixtral: Stop at EOS token (#183) devonthomas35 2023-12-23 21:25:42 -0800
2794f04873 Fix precommit hooks Devon Thomas 2023-12-23 20:00:01 -0800
5458722776 Fix precommit hooks Devon Thomas 2023-12-23 19:58:07 -0800
c79b004d27 Precommit format files Devon Thomas 2023-12-23 19:56:44 -0800
f2baee6fd2 Stop at EOS token Devon Thomas 2023-12-23 19:33:29 -0800
909a1e6909 fashion-mnist example (#180) Kashif Rasul 2023-12-23 16:34:45 +0100
0371d90ccb

fashion-mnist example (#180) Kashif Rasul 2023-12-23 16:34:45 +0100
0c2f1a3c55 fix from review Kashif Rasul 2023-12-23 16:25:53 +0100
ebadac12e8 use non-zero exit code on error (#177) Daniel Strobusch 2023-12-23 16:10:13 +0100
848f118ac5

use non-zero exit code on error (#177) Daniel Strobusch 2023-12-23 16:10:13 +0100
5320c4ff1f fix bad convert parameter (#178) Daniel Strobusch 2023-12-23 16:09:49 +0100
092e87211e

fix bad convert parameter (#178) Daniel Strobusch 2023-12-23 16:09:49 +0100
0ed2de6a61 fashion mnist example Kashif Rasul 2023-12-23 11:22:23 +0100
352b488cf6

fix bad convert parameter Daniel Strobusch 2023-12-23 09:05:45 +0100
668d0f3b08

use non-zero exit code on error Daniel Strobusch 2023-12-23 08:46:31 +0100
88df9691b3 Align CLI args and some smaller fixes (#167) Alvaro Bartolome 2023-12-22 23:34:32 +0100
f4709cb807

Align CLI args and some smaller fixes (#167) Alvaro Bartolome 2023-12-22 23:34:32 +0100
1abaebffdf one more Awni Hannun 2023-12-22 14:33:31 -0800
7296440151 standardize Awni Hannun 2023-12-22 14:32:23 -0800
d05fa79284 Align CLI args and minor fixes Alvaro Bartolome 2023-12-21 09:49:45 +0100
bdc1f4d1f6 Fix variable naming of config in mixtral/convert.py Alvaro Bartolome 2023-12-21 09:42:35 +0100
192ea3dd29 Add .DS_Store files to .gitignore Alvaro Bartolome 2023-12-21 09:38:38 +0100
7ca5bd7208 Fix conversion + inference errors. - Mistral (#176) Vaibhav Srivastav 2023-12-23 03:40:25 +0530
0eaa323c10

Fix conversion + inference errors. - Mistral (#176) Vaibhav Srivastav 2023-12-23 03:40:25 +0530
325d1b3ba9 wire rope_theta throuugh to nn.RoPE Awni Hannun 2023-12-22 14:09:45 -0800
adbd188c57 Fix conversion + inference errors. Vaibhav Srivastav 2023-12-23 01:19:58 +0530
ce7815b931 feat: add mistral tps (#173) Todsaporn Banjerdkit 2023-12-22 22:55:57 +0700
7ae445f6c7

feat: add mistral tps (#173) Todsaporn Banjerdkit 2023-12-22 22:55:57 +0700
31273bafbf eval params before timing + format Awni Hannun 2023-12-22 07:54:36 -0800
d7693b8f8a feat: add mistral tps katopz 2023-12-22 14:22:58 +0700
dc1a0ec67d stream conversion Awni Hannun 2023-12-21 15:25:51 -0800
e7b7c4fa92 fix typo (#169) Daniel Strobusch 2023-12-21 23:17:11 +0100
188a91074b

fix typo (#169) Daniel Strobusch 2023-12-21 23:17:11 +0100
e1d0c254d6 Quantize example (#162) Awni Hannun 2023-12-21 12:59:37 -0800
3cf436b529

Quantize example (#162) Awni Hannun 2023-12-21 12:59:37 -0800
942a6ef620 qwen conversion Awni Hannun 2023-12-21 12:54:47 -0800
9dbbd8755b mixtral Awni Hannun 2023-12-21 10:53:14 -0800
029e1f1ca4

fix typo Daniel Strobusch 2023-12-21 18:09:17 +0100
17d0967617 Add support for byt5 models (#161) Juarez Bochi 2023-12-21 11:46:36 -0500
4c9db80ed2

Add support for byt5 models (#161) Juarez Bochi 2023-12-21 11:46:36 -0500
9890ae6403

Remove unused import Juarez Bochi 2023-12-21 10:08:03 -0500
c66eed8e9f phi2 quantized Awni Hannun 2023-12-20 21:43:54 -0800
c8abd7906d llama / mistral conversion in good shape Awni Hannun 2023-12-20 21:25:25 -0800
aced530649 args for quantization Awni Hannun 2023-12-20 21:08:03 -0800
89db6ffdfe quantization in mistral / nits in llama Awni Hannun 2023-12-20 17:16:58 -0800
af4d82c93a one config processor Awni Hannun 2023-12-20 15:00:06 -0800
f6684df32a conversion + quantization working Awni Hannun 2023-12-20 14:48:30 -0800
8b824fb768 testing quantization Awni Hannun 2023-12-20 14:21:40 -0800
3914b149a5 update path to load weights (#164) Deven Mistry 2023-12-21 09:31:17 -0500
6c574dbecf

update path to load weights (#164) Deven Mistry 2023-12-21 09:31:17 -0500
be121ba181 updated results (#165) Sarthak Yadav 2023-12-21 15:30:17 +0100
4addd02988

updated results (#165) Sarthak Yadav 2023-12-21 15:30:17 +0100
18f0a96cee 1. Add user warning for sequences over 2048 tokens in iterate_batches. (#166) wyanzhao 2023-12-21 06:29:31 -0800
22620de3ee

1. Add user warning for sequences over 2048 tokens in iterate_batches. (#166) wyanzhao 2023-12-21 06:29:31 -0800
58f409feb0 rename --model_path to --model-path (#151) Daniel Strobusch 2023-12-21 15:28:57 +0100
43b6522af2

rename --model_path to --model-path (#151) Daniel Strobusch 2023-12-21 15:28:57 +0100
f93d0feb0b

rename --model_path to --model-path Daniel Strobusch 2023-12-20 10:35:34 +0100
d8a4920e66 1. Add user warning for sequences over 2048 tokens in iterate_batches. wyanzhao 2023-12-20 23:35:16 -0800
33e96625c8 updated results Sarthak Yadav 2023-12-21 08:07:41 +0100
043220f0b7 update path to load weights deven367 2023-12-20 23:08:06 -0500
df44fc5008 fix typo in readme (#163) Deven Mistry 2023-12-20 22:47:41 -0500
3efb1cc2cc

fix typo in readme (#163) Deven Mistry 2023-12-20 22:47:41 -0500
7c4d2ef4f4 fix typo in readme deven367 2023-12-20 22:43:04 -0500
9f3a83df63

Add support for byt5 models Juarez Bochi 2023-12-20 17:09:18 -0500
de2c1022e3 Removed debug message regarding orphaned weights Pawel Kowalski 2023-12-20 18:43:40 +0100
dac547367d Stable diffusion XL working; "add_embedding" layer not implemented Pawel Kowalski 2023-12-20 17:25:05 +0100
fe2291710f partial implementation of SD XL, incl. CLIP with projection, but doesn't produce good output Pawel Kowalski 2023-12-19 13:21:34 +0100
c63870d3cb partial implementation of SD XL, incl. CLIP with projection, but doesn't produce good output Pawel Kowalski 2023-12-19 13:21:07 +0100
e091fd2abc Reverting to a single model Pawel Kowalski 2023-12-14 09:57:55 +0100
b417c79673 moved the weight squeeze to map_unet_weights, style check Pawel Kowalski 2023-12-13 23:36:47 +0100
49cbb24b49 Allow integer as attention_head_dim, reshape downloaded weights to match model if mismatch Pawel Kowalski 2023-12-12 22:23:52 +0100
6c13c072b2 Use config.json in llama (#159) Pedro Cuenca 2023-12-20 19:34:44 +0100
ce30cc3d8f

Use config.json in llama (#159) Pedro Cuenca 2023-12-20 19:34:44 +0100
1120adab98 Merge remote-tracking branch 'upstream/main' into llama-config-json Pedro Cuenca 2023-12-20 19:23:38 +0100
b3f23a7191 Add llms subdir + update README (#145) Awni Hannun 2023-12-20 10:22:25 -0800
27c0a8c002

Add llms subdir + update README (#145) Awni Hannun 2023-12-20 10:22:25 -0800
4dd4b670b3 Typo Pedro Cuenca 2023-12-20 19:21:37 +0100
5c29379c86 Fix convert Pedro Cuenca 2023-12-20 19:20:59 +0100
89280be600 Fix pop Pedro Cuenca 2023-12-20 19:20:09 +0100
03824e3477 format Awni Hannun 2023-12-20 10:14:01 -0800
7b5787e62c update readmes a bit Awni Hannun 2023-12-20 10:09:03 -0800
e4777e6698 use same pre-commit as mlx Awni Hannun 2023-12-19 09:30:32 -0800
3b98fb3a0f Fix minor typos, add some annotations Siddharth Mishra-Sharma 2023-12-20 13:10:05 -0500
d2422f0911 nits Awni Hannun 2023-12-19 09:05:15 -0800
f2216965c1 add llms subdir + update README Awni Hannun 2023-12-19 09:03:26 -0800
3c64f1a1dc Add config.json to Mixtral. (#158) Vaibhav Srivastav 2023-12-20 23:17:23 +0530
aed14618ca

Add config.json to Mixtral. (#158) Vaibhav Srivastav 2023-12-20 23:17:23 +0530
3ef1e9944a Use config.json in llama Pedro Cuenca 2023-12-20 18:09:28 +0100
b970e281ac

Update mixtral/mixtral.py Vaibhav Srivastav 2023-12-20 22:37:53 +0530
f881821895 Add config.json to Mixtral. Vaibhav Srivastav 2023-12-20 22:32:23 +0530
920bd13668 Use config.json, add model_type (#157) Pedro Cuenca 2023-12-20 17:39:37 +0100