Hugging Face's logo

Spaces:
Fynd
/
cloth-vton
Runtime error

App Files Community
abrarlohia commited on
Commit
cde67e2
·
verified ·
1 Parent(s): c79133e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ device = torch.device("cuda")
17
  torch_dtype = torch.bfloat16 # torch.float16
18
 
19
  def load_models(device=device, torch_dtype=torch_dtype,group_offloading=False):
20
- bfl_repo = "black-forest-labs/FLUX.1-dev"
21
  # Enable memory efficient attention
22
  text_encoder = CLIPTextModel.from_pretrained(bfl_repo, subfolder="text_encoder", torch_dtype=torch_dtype,)
23
  text_encoder_2 = T5EncoderModel.from_pretrained(bfl_repo, subfolder="text_encoder_2", torch_dtype=torch_dtype,)
 
17
  torch_dtype = torch.bfloat16 # torch.float16
18
 
19
  def load_models(device=device, torch_dtype=torch_dtype,group_offloading=False):
20
+ bfl_repo = "Fynd/flux-dev-1-clone"
21
  # Enable memory efficient attention
22
  text_encoder = CLIPTextModel.from_pretrained(bfl_repo, subfolder="text_encoder", torch_dtype=torch_dtype,)
23
  text_encoder_2 = T5EncoderModel.from_pretrained(bfl_repo, subfolder="text_encoder_2", torch_dtype=torch_dtype,)