Add new model

model/models/huggingface_models.py

@@ -28,10 +28,9 @@ def load_huggingface_model(model_name, model_type):
 
 
 if __name__ == "__main__":
-    # for name in ["SD-turbo", "SDXL-turbo"]: #"SD-turbo", "SDXL-turbo"
-    #     pipe = load_huggingface_model(name, "text2image")
+    for name in ["SD-turbo", "SDXL-turbo"]: #"SD-turbo", "SDXL-turbo"
+        pipe = load_huggingface_model(name, "text2image")
 
     # for name in ["IF-I-XL-v1.0"]:
     #     pipe = load_huggingface_model(name, 'text2image')
-    # pipe = DiffusionPipeline.from_pretrained("DeepFloyd/IF-I-XL-v1.0", variant="fp16", torch_dtype=torch.float16)
-
+    # pipe = DiffusionPipeline.from_pretrained("DeepFloyd/IF-I-XL-v1.0", variant="fp16", torch_dtype=torch.float16)