Spaces:
Running
on
Zero
Running
on
Zero
Yuxuan Zhang
commited on
Commit
·
9893975
1
Parent(s):
3ba8788
req
Browse files- app.py +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,
|
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/
|
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
|