Spaces:

lucalp
/

blt-entropy-patcher

Running on Zero

lucalp commited on 3 days ago

Commit

5c58fd6

1 Parent(s): 0d61fe6

Updating to use new args file

Files changed (1) hide show

app.py CHANGED Viewed

@@ -86,7 +86,7 @@ class BytelatentProcessor:
                 # 2. Load Bytelatent model components
                 consolidated_path = os.path.join(self.weights_dir, model_name)
-                train_args_path = os.path.join(consolidated_path, "params.json")
                 entropy_model_dir = os.path.join(consolidated_path, "entropy_model")
                 if not os.path.exists(train_args_path):

                 # 2. Load Bytelatent model components
                 consolidated_path = os.path.join(self.weights_dir, model_name)
+                train_args_path = os.path.join(consolidated_path, "train_args.json")
                 entropy_model_dir = os.path.join(consolidated_path, "entropy_model")
                 if not os.path.exists(train_args_path):