John6666 commited on
Commit
bf5fb76
β€’
1 Parent(s): 48ccfbd

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. genimage.py +1 -1
  3. requirements.txt +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: πŸ‘€πŸ˜»
4
  colorFrom: red
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 4.38.1
8
  app_file: app.py
9
  pinned: true
10
  license: apache-2.0
 
4
  colorFrom: red
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 4.39.0
8
  app_file: app.py
9
  pinned: true
10
  license: apache-2.0
genimage.py CHANGED
@@ -7,7 +7,7 @@ def load_pipeline():
7
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
8
  pipe = AutoPipelineForText2Image.from_pretrained(
9
  "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
10
- custom_pipeline="lpw_stable_diffusion",
11
  torch_dtype=torch.float16,
12
  )
13
  pipe.to(device)
 
7
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
8
  pipe = AutoPipelineForText2Image.from_pretrained(
9
  "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
10
+ custom_pipeline="lpw_stable_diffusion_xl",
11
  torch_dtype=torch.float16,
12
  )
13
  pipe.to(device)
requirements.txt CHANGED
@@ -13,5 +13,5 @@ dartrs
13
  httpx==0.13.3
14
  httpcore
15
  googletrans==4.0.0rc1
16
- git+https://github.com/huggingface/diffusers
17
  rapidfuzz
 
13
  httpx==0.13.3
14
  httpcore
15
  googletrans==4.0.0rc1
16
+ diffusers
17
  rapidfuzz