feat: 新增生图模型

hf_down.py

@@ -5,7 +5,7 @@ os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
 
 model_list = [
     "openai-community/gpt2",
-    # "stabilityai/stable-diffusion-xl-base-1.0"
+    "CompVis/stable-diffusion-v1-4"
 ]
 
 os.system("pip install -U huggingface-hub")
@@ -14,4 +14,4 @@ for model_path in model_list:
     root_dir = os.path.basename(model_path)
     os.system(
         f"huggingface-cli download --resume-download  {model_path} --local-dir ./{root_dir}/{model_path} \
-        --local-dir-use-symlinks False --token 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")
+        --local-dir-use-symlinks False")

stable-diffusion-v1-4/infer.py

+import torch
+from diffusers import StableDiffusionPipeline
+
+model_id = "CompVis/stable-diffusion-v1-4"
+device = "cuda"
+
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to(device)
+
+prompt = "a photo of an astronaut riding a horse on mars"
+image = pipe(prompt).images[0]
+
+image.save("output.png")
+print("The output image has been saved as output.png")