akhaliq HF Staff commited on
Commit
b2c1288
·
verified ·
1 Parent(s): eeb1782

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ if hf_token:
11
  login(token=hf_token)
12
 
13
  # Load the model with token
14
- pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.2", torch_dtype=torch.bfloat16, token=hf_token)
15
  pipe.to(device="cuda")
16
 
17
  # ZeroGPU AoT compilation
 
11
  login(token=hf_token)
12
 
13
  # Load the model with token
14
+ pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.2", torch_dtype=torch.bfloat16, use_auth_token=hf_token)
15
  pipe.to(device="cuda")
16
 
17
  # ZeroGPU AoT compilation