Upload app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ IMAGE_SIZE = 1024
|
|
30 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
31 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
32 |
|
33 |
-
|
34 |
|
35 |
#bfl_repo="black-forest-labs/FLUX.1-dev"
|
36 |
bfl_repo="camenduru/FLUX.1-dev-diffusers"
|
|
|
30 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
31 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
32 |
|
33 |
+
if HF_TOKEN: login(token=HF_TOKEN)
|
34 |
|
35 |
#bfl_repo="black-forest-labs/FLUX.1-dev"
|
36 |
bfl_repo="camenduru/FLUX.1-dev-diffusers"
|