Hugging Face's logo

Spaces:
briaai
/
Multi-Image-Prompt
Running on Zero

App Files Community
mokady commited on
Commit
1e992f3
·
verified ·
1 Parent(s): 6e51693

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ hf_token = os.environ.get("HF_TOKEN")
10
  from diffusers import AutoPipelineForText2Image
11
 
12
 
13
- device = "cuda" if torch.cuda.is_available() else "cpu"
14
  pipe = AutoPipelineForText2Image.from_pretrained("briaai/BRIA-2.3", torch_dtype=torch.float16, force_zeros_for_empty_prompt=False).to(device)
15
  pipe.load_ip_adapter("briaai/Image-Prompt", subfolder='models', weight_name="ip_adapter_bria.bin")
16
 
 
10
  from diffusers import AutoPipelineForText2Image
11
 
12
 
13
+ device = "cuda" #if torch.cuda.is_available() else "cpu"
14
  pipe = AutoPipelineForText2Image.from_pretrained("briaai/BRIA-2.3", torch_dtype=torch.float16, force_zeros_for_empty_prompt=False).to(device)
15
  pipe.load_ip_adapter("briaai/Image-Prompt", subfolder='models', weight_name="ip_adapter_bria.bin")
16