File size: 2,504 Bytes
6c27fdd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
import sys
model_name = sys.argv[1]
model_card = f"""---
language:
- en
license: openrail++
tags:
- stable-diffusion
- stable-diffusion-diffusers
- stable-diffusion-xl
- text-to-image
- art
- artistic
- diffusers
- anime
---
# {model_name.split("/")[-1].replace("-", " ").capitalize()}
`{model_name}` is a Stable Diffusion model that has been fine-tuned on [stabilityai/stable-diffusion-xl-base-1.0](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0).
Please consider supporting me:
- on [Patreon](https://www.patreon.com/Lykon275)
- or [buy me a coffee](https://snipfeed.co/lykon)
## Diffusers
For more general information on how to run text-to-image models with 🧨 Diffusers, see [the docs](https://huggingface.co/docs/diffusers/using-diffusers/conditional_image_generation).
1. Installation
```
pip install diffusers transformers accelerate
```
2. Run
```py
from diffusers import AutoPipelineForText2Image, DEISMultistepScheduler
import torch
pipe = AutoPipelineForText2Image.from_pretrained('{model_name}', torch_dtype=torch.float16, variant="fp16")
pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
pipe = pipe.to("cuda")
prompt = "portrait photo of muscular bearded guy in a worn mech suit, light bokeh, intricate, steel metal, elegant, sharp focus, soft lighting, vibrant colors"
generator = torch.manual_seed(0)
image = pipe(prompt, num_inference_steps=25).images[0]
image.save("./image.png")
```
"""
from huggingface_hub import HfApi
api = HfApi()
read_me_path = "./README.md"
with open(read_me_path, "w") as f:
f.write(model_card)
api.upload_file(
path_or_fileobj=read_me_path,
path_in_repo=read_me_path,
repo_id=model_name,
repo_type="model",
)
from diffusers import AutoPipelineForText2Image, DEISMultistepScheduler
import torch
pipe = AutoPipelineForText2Image.from_pretrained(model_name, torch_dtype=torch.float16)
pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
pipe = pipe.to("cuda")
prompt = "portrait photo of muscular bearded guy in a worn mech suit, light bokeh, intricate, steel metal, elegant, sharp focus, soft lighting, vibrant colors"
generator = torch.manual_seed(0)
image = pipe(prompt, num_inference_steps=25).images[0]
image_path = "./image.png"
image.save(image_path)
api.upload_file(
path_or_fileobj=image_path,
path_in_repo=image_path,
repo_id=model_name,
repo_type="model",
)
pipe.push_to_hub(model_name, variant="fp16")
|