tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updated readme

Browse Source
This commit is contained in:
Brett Kuprel
2022-06-30 16:50:04 -04:00
parent 2311a1af7b
commit 08b158d580
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
min_dalle/load_params.py
+1 -1
View File
@@ -107,7 +107,7 @@ def convert_dalle_bart_torch_from_flax_params(
return P
def convert_and_save_mega_torch_params(is_mega: bool, model_path: str):
def convert_and_save_torch_params(is_mega: bool, model_path: str):
print("converting params to torch")
layer_count = 24 if is_mega else 12
flax_params = load_dalle_bart_flax_params(model_path)