Update app.py

app.py

@@ -27,7 +27,7 @@ from modeling.qwen2 import Qwen2Tokenizer
 from huggingface_hub import snapshot_download
 
 save_dir = "./model_weights"
-repo_id = "ByteDance-Seed/BAGEL-7B-MoT"
+repo_id = "sanaka87/BAGEL-ReAlign"
 cache_dir = save_dir + "/cache"
 
 snapshot_download(
@@ -109,7 +109,7 @@ else:
             
 model = load_checkpoint_and_dispatch(
     model,
-    checkpoint=os.path.join(model_path, "ema.safetensors"),
+    checkpoint=os.path.join(model_path, "model_bf16.safetensors"),
     device_map=device_map,
     offload_buffers=True,
     offload_folder="offload",