simon5u

state dict length is one of 102, 182, 267, 522?

13

Hi, may I know how to use the get_small_model.py? I tried to repeat the resnet50 pruning_train.py on imagenet and get the resnet50 checkpoint, then i load the model using get_small_model.py....

How to get the summary if the model output consists of int and str?

1

**Describe the bug** torchinfo.py", line 448, in traverse_input_data result = aggregate( TypeError: unsupported operand type(s) for +: 'int' and 'str' It seems like the torchinfo.py cannot mix the different model...

any example for stable diffusion?

1

Hi, could you please share an example on how to use mlx-swift to run the stable diffusion 1.5, which is similar to the one given under https://github.com/ml-explore/mlx-examples/tree/main/stable_diffusion?

How to modify or update the model weight in swift?

2

Hi, we are developing a swift app in ipad where we managed to load the stable diffusion model into the app, but we want to partially change the model weight...