Update gen.py

gen.py

@@ -20,20 +20,26 @@ def download_pretrained_models(
     ckpt_path: str,
     tokenizer_path: str
 ):
+    print("creating local directories...")
     os.makedirs(llama_weight_path)
     os.makedirs(tokenizer_weight_path)
 
+    print("initialize GCS client...")
     storage_client = storage.Client.create_anonymous_client()
     bucket = storage_client.bucket(bucket_name)
 
+    print(f"download {ckpt_path} model weights...")
     blobs = bucket.list_blobs(prefix=f"{ckpt_path}/")
     for blob in blobs:
         filename = blob.name.split("/")[1]
+        print(f"-{filename}")
         blob.download_to_filename(f"{llama_weight_path}/{filename}")
 
+    print(f"download {tokenizer_path} tokenizer weights...")
     blobs = bucket.list_blobs(prefix=f"{tokenizer_path}/")
     for blob in blobs:
         filename = blob.name.split("/")[1]
+        print(f"-{filename}")
         blob.download_to_filename(f"{tokenizer_weight_path}/{filename}")    
 
 def get_pretrained_models(