Hugging Face's logo

Spaces:
mokady
/
bria-32
Runtime error

App Files Community
mokady commited on
Commit
03e5e5c
·
verified ·
1 Parent(s): de94a01

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
 
12
 
13
  MODEL_ID = "briaai/BRIA-3.2"
14
- pipe = BriaPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16, use_auth_token=hf_token))
15
  pipe.to("cuda")
16
 
17
  @spaces.GPU(duration=1500)
 
11
 
12
 
13
  MODEL_ID = "briaai/BRIA-3.2"
14
+ pipe = BriaPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16, use_auth_token=hf_token)
15
  pipe.to("cuda")
16
 
17
  @spaces.GPU(duration=1500)