sanjeevbora commited on
Commit
f023f9c
·
verified ·
1 Parent(s): ec646e7

updated with my model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ token = os.getenv("HUGGINGFACE_HUB_TOKEN")
11
  login(token=token)
12
 
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
- model_repo_id = "black-forest-labs/FLUX.1-dev" #Replace to the model you would like to use
15
 
16
  if torch.cuda.is_available():
17
  torch_dtype = torch.float16
 
11
  login(token=token)
12
 
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
+ model_repo_id = "sanjeevbora/flux.1_dev_ebiz"
15
 
16
  if torch.cuda.is_available():
17
  torch_dtype = torch.float16