fix black check

01-ai · Jan 16, 2024 · 9eb2c35 · 9eb2c35
1 parent 9ff30d0
commit 9eb2c35
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/demo/web_demo.py b/demo/web_demo.py
@@ -209,7 +209,10 @@ def user(query, history):
         device_map = "auto"
 
     model = AutoModelForCausalLM.from_pretrained(
-        args.checkpoint_path, device_map=device_map, torch_dtype="auto", trust_remote_code=True
+        args.checkpoint_path,
+        device_map=device_map,
+        torch_dtype="auto",
+        trust_remote_code=True,
     ).eval()
 
     main(args)