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)]
|