Update src/linfusion/linfusion.py

src/linfusion/linfusion.py

@@ -6,8 +6,8 @@ from .attention import GeneralizedLinearAttention
 
 
 model_dict = {
-    "runwayml/stable-diffusion-v1-5": "Yuanshi/LinFusion-1-5",
-    "stablediffusionapi/realistic-vision-v51": "Yuanshi/LinFusion-1-5",
+    "CompVis/stable-diffusion-v1-4": "Yuanshi/LinFusion-1-5",
+    "SG161222/Realistic_Vision_V5.1_noVAE": "Yuanshi/LinFusion-1-5",
     "Lykon/dreamshaper-8": "Yuanshi/LinFusion-1-5",
 }