Dmitry Trifonov commited on
Commit
bdc1919
·
1 Parent(s): 6c7a4d8

use dreamshaper v8

Browse files
Files changed (2) hide show
  1. app.py +5 -7
  2. text_to_image.py +6 -4
app.py CHANGED
@@ -150,13 +150,12 @@ with block:
150
  "
151
  >
152
  <h1 style="font-weight: 900; margin-bottom: 7px;">
153
- SDXL Turbo by Fair Compute
154
  </h1>
155
  </div>
156
  <p style="margin-bottom: 10px; font-size: 94%">
157
- SDXL Turbo model for generating high-quality images.<br/>
158
- Model: <a href="https://huggingface.co/stabilityai/sdxl-turbo">https://huggingface.co/stabilityai/sdxl-turbo</a><br/>
159
- <br/>
160
  </p>
161
  </div>
162
  """
@@ -191,9 +190,8 @@ with block:
191
 
192
  gr.HTML(
193
  """
194
- <div class="acknowledgments">
195
- Run AI models on your computer and avoid paying egregious cloud fees.
196
- This demo is powered by the <a href="https://faircompute.com/">FairCompute</a> platform.
197
  </div>
198
  """
199
  )
 
150
  "
151
  >
152
  <h1 style="font-weight: 900; margin-bottom: 7px;">
153
+ DreamShaper by FairCompute
154
  </h1>
155
  </div>
156
  <p style="margin-bottom: 10px; font-size: 94%">
157
+ DreamShaper V8 model for generating high-quality images.<br/>
158
+ Model: <a href="https://huggingface.co/Lykon/dreamshaper-8">https://huggingface.co/Lykon/dreamshaper-8</a><br/>
 
159
  </p>
160
  </div>
161
  """
 
190
 
191
  gr.HTML(
192
  """
193
+ <div class="acknowledgments">
194
+ <p>Run AI models on your home computers, powered by <a href="https://faircompute.com/" style="text-decoration: underline;">FairCompute</a></p>
 
195
  </div>
196
  """
197
  )
text_to_image.py CHANGED
@@ -17,7 +17,7 @@ TARGET_NODE = "119eccba-2388-43c1-bdb9-02133049604c"
17
  # ENDPOINT_ADDRESS = "http://localhost:5000"
18
  # TARGET_NODE = "ef09913249aa40ecba7d0097f7622855"
19
 
20
- DOCKER_IMAGE = "faircompute/diffusers-api-sdxl-turbo"
21
 
22
 
23
  class EndpointClient:
@@ -30,14 +30,16 @@ class EndpointClient:
30
  inputs = {
31
  "modelInputs": {
32
  "prompt": prompt,
33
- "num_inference_steps": 4,
34
- "guidance_scale": 0.0,
35
  "width": 512,
36
  "height": 512,
37
  },
38
  "callInputs": {
39
- "MODEL_ID": "stabilityai/sdxl-turbo",
40
  "PIPELINE": "AutoPipelineForText2Image",
 
 
 
41
  },
42
  }
43
 
 
17
  # ENDPOINT_ADDRESS = "http://localhost:5000"
18
  # TARGET_NODE = "ef09913249aa40ecba7d0097f7622855"
19
 
20
+ DOCKER_IMAGE = "faircompute/diffusers-api-dreamshaper-8"
21
 
22
 
23
  class EndpointClient:
 
30
  inputs = {
31
  "modelInputs": {
32
  "prompt": prompt,
33
+ "num_inference_steps": 25,
 
34
  "width": 512,
35
  "height": 512,
36
  },
37
  "callInputs": {
38
+ "MODEL_ID": "lykon/dreamshaper-8",
39
  "PIPELINE": "AutoPipelineForText2Image",
40
+ "SCHEDULER": "DEISMultistepScheduler",
41
+ "PRECISION": "fp16",
42
+ "REVISION": "fp16",
43
  },
44
  }
45