Views
No views yet
1import torch
2
3# Load FP16 state dict
4state_dict = torch.load("mp_rank_00_model_states_fp16.pt", map_location="cpu")
5
6# Load into model (model must be converted to FP16 first)
7model = YourResembleEnhanceModel()
8model = model.half() # Convert to FP16
9model.load_state_dict(state_dict)
10model.eval()1import coremltools as ct
2
3# Convert PyTorch model to CoreML
4mlmodel = ct.convert(
5 model,
6 inputs=[ct.TensorType(name="input", shape=input_shape)],
7 minimum_deployment_target=ct.target.iOS16
8)
9
10# Save as .mlmodel
11mlmodel.save("ResembleEnhanceFP16.mlmodel")