Skip to content

Commit 182b9d4

Browse files
committed
fix: put gcp to None when non bucket is pass
1 parent d2db30b commit 182b9d4

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/genesys/generate.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,7 @@ def check_batch_size(self):
3131

3232

3333
def main(config: Config):
34-
if config.gcp_bucket is not None:
35-
gcp_bucket = GcpBucket(config.gcp_bucket)
34+
gcp_bucket = GcpBucket(config.gcp_bucket) if config.gcp_bucket is not None else None
3635

3736
llm = sgl.Engine(model_path=config.name_model, tp_size=config.num_gpus)
3837
tokenizer = AutoTokenizer.from_pretrained(config.name_model)

0 commit comments

Comments
 (0)