Update README.md

README.md

@@ -9,19 +9,16 @@ tags:
 - art
 - tensorRT
 ---
-# Model Card for fast_sd_1.4
+# Model Card for fast_sd_1.5
 
-fast_sd_1.4 is a tensorRT implementation of Stable diffusion model, with stable 1.4 weights. Main features 
+fast_sd_1.5 is a tensorRT implementation of Stable diffusion model, with stable 1.5 weights. Main features 
 
 - weights: original stable diffusion 1.4 weights
-- input image size range: 100x100 ~ 200x200
-- text encoder: clipxxx
-- super-resultion: 4x by default.
-- controlnet
-  - canny: yes
-  - depth(midas): yes
-  - xxx: yes
- 
+- input image size range: 256~640 （min 256x256, max 640x640）
+- device requirements: Nvidia Ampere architecture or newer (A100)
+- super-resultion: 4x by default, optional.
+
+fast_sd_1.5 is a part of TME lyralab Fast Model Plan.
 
 
 ### Model Sources [optional]
@@ -30,32 +27,21 @@ fast_sd_1.4 is a tensorRT implementation of Stable diffusion model, with stable
 
 ## Uses
 
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
-
-### Direct Use
-
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
-
-[More Information Needed]
-
-
-
-### Recommendations
-
-
-## How to Get Started with the Model
-
-Use the code below to get started with the model.
+```python
+from fast_sd_1.5 import FSD
+from PIL import Image
 
-[More Information Needed]
 
+inference_model = FSD(plan="path/to/plan/dir", superres="4x")
 
-## Glossary [optional]
+# text input
+text = "A red ballon flying in the sky"
+output_img = inference_model.infer(text=text)
 
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+# img input
+img = Image.open("test.jpg")
+output_img = inference_model.infer(img=img)
+```
 
-[More Information Needed]
 
-## More Information [optional]
 
-[More Information Needed]