dataautogpt3 commited on
Commit
e257ba6
1 Parent(s): 903173e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +35 -4
README.md CHANGED
@@ -89,11 +89,42 @@ best quality, HD, ~*~aesthetic~*~,
89
 
90
  ## Use it with 🧨 diffusers
91
  ```python
92
- from diffusers import AutoPipelineForText2Image
93
  import torch
94
-
95
- pipeline = AutoPipelineForText2Image.from_pretrained('dataautogpt3/ProteusV0.1', torch_dtype=torch.float16).to('cuda')
96
- image = pipeline('black fluffy gorgeous dangerous cat animal creature, large orange eyes, big fluffy ears, piercing gaze, full moon, dark ambiance, best quality, extremely detailed').images[0]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
97
  ```
98
 
99
  please support the work I do through donating to me on:
 
89
 
90
  ## Use it with 🧨 diffusers
91
  ```python
 
92
  import torch
93
+ from diffusers import (
94
+ StableDiffusionXLPipeline,
95
+ KDPM2AncestralDiscreteScheduler,
96
+ AutoencoderKL
97
+ )
98
+
99
+ # Load VAE component
100
+ vae = AutoencoderKL.from_pretrained(
101
+ "madebyollin/sdxl-vae-fp16-fix",
102
+ torch_dtype=torch.float16
103
+ )
104
+
105
+ # Configure the pipeline
106
+ pipe = StableDiffusionXLPipeline.from_pretrained(
107
+ "dataautogpt3/ProteusV0.1",
108
+ vae=vae,
109
+ torch_dtype=torch.float16,
110
+ use_safetensors=True,
111
+ variant="fp16"
112
+ )
113
+ pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
114
+ pipe.to('cuda')
115
+
116
+ # Define prompts and generate image
117
+ prompt = "black fluffy gorgeous dangerous cat animal creature, large orange eyes, big fluffy ears, piercing gaze, full moon, dark ambiance, best quality, extremely detailed"
118
+ negative_prompt = "nsfw, bad quality, bad anatomy, worst quality, low quality, low resolutions, extra fingers, blur, blurry, ugly, wrongs proportions, watermark, image artifacts, lowres, ugly, jpeg artifacts, deformed, noisy image"
119
+
120
+ image = pipe(
121
+ prompt,
122
+ negative_prompt=negative_prompt,
123
+ width=1024,
124
+ height=1024,
125
+ guidance_scale=7,
126
+ num_inference_steps=20
127
+ ).images[0]
128
  ```
129
 
130
  please support the work I do through donating to me on: