We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d2db30b commit 182b9d4Copy full SHA for 182b9d4
src/genesys/generate.py
@@ -31,8 +31,7 @@ def check_batch_size(self):
31
32
33
def main(config: Config):
34
- if config.gcp_bucket is not None:
35
- gcp_bucket = GcpBucket(config.gcp_bucket)
+ gcp_bucket = GcpBucket(config.gcp_bucket) if config.gcp_bucket is not None else None
36
37
llm = sgl.Engine(model_path=config.name_model, tp_size=config.num_gpus)
38
tokenizer = AutoTokenizer.from_pretrained(config.name_model)
0 commit comments