Commit ·
c8514e2
1
Parent(s): cf81de1
removed seed in generate image
Browse files
Gemini_Musicgen (3).ipynb → Gemini_Musicgen.ipynb
RENAMED
|
File without changes
|
app.py
CHANGED
|
@@ -398,7 +398,7 @@ def generate_image(prompt,num_inference_steps=1, guidance_scale=3, seed=None):
|
|
| 398 |
print(f"Random Seed: {seed}")
|
| 399 |
|
| 400 |
start_time = time.time()
|
| 401 |
-
image = pipe(prompt=prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale
|
| 402 |
end_time = time.time()
|
| 403 |
inference_time = end_time - start_time
|
| 404 |
print(f" Inference Time for Cover Generation: {inference_time}")
|
|
|
|
| 398 |
print(f"Random Seed: {seed}")
|
| 399 |
|
| 400 |
start_time = time.time()
|
| 401 |
+
image = pipe(prompt=prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).images[0]
|
| 402 |
end_time = time.time()
|
| 403 |
inference_time = end_time - start_time
|
| 404 |
print(f" Inference Time for Cover Generation: {inference_time}")
|