| 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") | |