Update app.py
Browse files
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,
|
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
|