Upload folder using huggingface_hub

README.md

@@ -36,12 +36,17 @@ tokenizer.save_pretrained(save_path)
 model = transformers.AutoModelForCausalLM.from_config(config, torch_dtype=torch.bfloat16)
 model.generation_config = transformers.GenerationConfig.from_pretrained(model_id)
 
+transformers.set_seed(42)
+with torch.no_grad():
+    for _, p in sorted(model.named_parameters()):
+        torch.nn.init.uniform_(p, -0.1, 0.1)
+
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, do_sample=False, device='cuda')
 print(pipe('Hello World!'))
 
 model.save_pretrained(save_path)
 
 os.system(f'ls -alh {save_path}')
-# create_repo(repo_id, exist_ok=True)
-# upload_folder(repo_id=repo_id, folder_path=save_path)
+create_repo(repo_id, exist_ok=True)
+upload_folder(repo_id=repo_id, folder_path=save_path)
 ```