Spaces:
Running
Running
Commit
Β·
d66da66
1
Parent(s):
cff0372
Fix typo in prompt generation message for clarity
Browse files
medvqa/submission_samples/gi-2025/submission_task2.py
CHANGED
|
@@ -85,7 +85,7 @@ print(f"π Output folder: {os.path.abspath(output_folder)}")
|
|
| 85 |
|
| 86 |
batch_size = 2 # Adjust based on your GPU memory, number of prompts to generate in one go
|
| 87 |
print(
|
| 88 |
-
f"π We have {len(test_prompts)} prompts and we are generating for
|
| 89 |
for i in tqdm(range(0, len(test_prompts), batch_size), desc="π Generating images"):
|
| 90 |
batch = test_prompts[i:i + batch_size]
|
| 91 |
batched_prompts = [p for p in batch for _ in range(num_per_prompt)]
|
|
|
|
| 85 |
|
| 86 |
batch_size = 2 # Adjust based on your GPU memory, number of prompts to generate in one go
|
| 87 |
print(
|
| 88 |
+
f"π We have {len(test_prompts)} prompts and we are generating for {batch_size} prompts at once. ")
|
| 89 |
for i in tqdm(range(0, len(test_prompts), batch_size), desc="π Generating images"):
|
| 90 |
batch = test_prompts[i:i + batch_size]
|
| 91 |
batched_prompts = [p for p in batch for _ in range(num_per_prompt)]
|