modify model card

README.md

@@ -16,23 +16,23 @@ This repository hosts the pretrained parameters for the SuperMat project, as des
 
 | Model | Description |
 | --- | --- |
-| supermat.ckpt | Base SuperMat model for material decomposition
-| supermat_mv.ckpt | Multi-view version of SuperMat processing six orthogonal views
-| uv_refine_bc | UV refinement network for albedo materials
-| uv_refine_rm | UV refinement network for roughness & metallic materials
+| supermat.pth | Base SuperMat model for material decomposition
+| supermat_mv.pth | Multi-view version of SuperMat processing six orthogonal views
+| uv_refine_bc.pth | UV refinement network for albedo materials
+| uv_refine_rm.pth | UV refinement network for roughness & metallic materials
 
 All models are built upon the base model `stabilityai/stable-diffusion-2-1`.
 Note: The official `stabilityai/stable-diffusion-2-1` model has been removed. You may need to obtain the base model parameters through alternative sources, such as `sd2-community/stable-diffusion-2-1`.
 
 # Model Details
-#### SuperMat (supermat.ckpt)
+#### SuperMat (supermat.pth)
 The core model for material decomposition. It takes RGBA images as input and decomposes materials from the target object.
-#### SuperMat Multi-View (supermat_mv.ckpt)
+#### SuperMat Multi-View (supermat_mv.pth)
 An extended version that processes six orthogonal views simultaneously. This model leverages multi-view consistency for improved material estimation. For each view, the camera-to-world (c2w) matrix is provided as camera embeddings.
 #### UV Refinement Networks
 Two specialized networks for refining UV maps:
-- uv_refine_bc: Refines the UV map for albedo materials
-- uv_refine_rm: Refines the UV map for roughness & metallic materials
+- uv_refine_bc.pth: Refines the UV map for albedo materials
+- uv_refine_rm.pth: Refines the UV map for roughness & metallic materials
 
 # Download & Usage
 Download the desired model(s) from this repository and place them in the checkpoints folder: