Yuxuan Zhang commited on
Commit
9893975
·
1 Parent(s): 3ba8788
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -120,12 +120,12 @@ def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_infer
120
  images = pipe(
121
  prompt=prompt,
122
  guidance_scale=guidance_scale,
123
- num_images_per_prompt=num_images, # 生成 num_images 张图
124
  num_inference_steps=num_inference_steps,
125
  width=width,
126
  height=height,
127
  generator=torch.Generator().manual_seed(seed)
128
- ).images # 获取生成的图片列表
129
 
130
  return images, seed
131
 
 
120
  images = pipe(
121
  prompt=prompt,
122
  guidance_scale=guidance_scale,
123
+ num_images_per_prompt=num_images,
124
  num_inference_steps=num_inference_steps,
125
  width=width,
126
  height=height,
127
  generator=torch.Generator().manual_seed(seed)
128
+ ).images
129
 
130
  return images, seed
131
 
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  accelerate>=1.3.0
2
  sentencepiece>=0.2.0
3
- git+https://github.com/zRzRzRzRzRzRzR/diffusers.git@glm
4
  gradio>=5.15.0
5
  transformers>=4.48.3
6
  openai>=1.61.0
 
1
  accelerate>=1.3.0
2
  sentencepiece>=0.2.0
3
+ git+https://github.com/huggingface/diffusers.git
4
  gradio>=5.15.0
5
  transformers>=4.48.3
6
  openai>=1.61.0