Spaces:
Running
on
Zero
Running
on
Zero
lifeng
commited on
Commit
·
675ceb3
1
Parent(s):
10afe14
修改参数
Browse files- README.md +1 -1
- app.py +7 -0
- requirements.txt +2 -1
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
title:
|
3 |
emoji: 🦀
|
4 |
colorFrom: blue
|
5 |
colorTo: blue
|
|
|
1 |
---
|
2 |
+
title: Happykala Magic Ghibli
|
3 |
emoji: 🦀
|
4 |
colorFrom: blue
|
5 |
colorTo: blue
|
app.py
CHANGED
@@ -12,6 +12,13 @@ from src.pipeline import FluxPipeline
|
|
12 |
from src.transformer_flux import FluxTransformer2DModel
|
13 |
from src.lora_helper import set_single_lora, set_multi_lora, unset_lora
|
14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
# Initialize the image processor
|
16 |
base_path = "black-forest-labs/FLUX.1-dev"
|
17 |
lora_base_path = "./models"
|
|
|
12 |
from src.transformer_flux import FluxTransformer2DModel
|
13 |
from src.lora_helper import set_single_lora, set_multi_lora, unset_lora
|
14 |
|
15 |
+
token = os.getenv("hugface_token")
|
16 |
+
if token:
|
17 |
+
login(token=token)
|
18 |
+
print("Login successful!")
|
19 |
+
else:
|
20 |
+
print("hugface_token not found in environment variables.")
|
21 |
+
|
22 |
# Initialize the image processor
|
23 |
base_path = "black-forest-labs/FLUX.1-dev"
|
24 |
lora_base_path = "./models"
|
requirements.txt
CHANGED
@@ -11,4 +11,5 @@ protobuf==5.29.3
|
|
11 |
requests==2.32.3
|
12 |
safetensors==0.5.2
|
13 |
sentencepiece==0.2.0
|
14 |
-
transformers==4.49.0
|
|
|
|
11 |
requests==2.32.3
|
12 |
safetensors==0.5.2
|
13 |
sentencepiece==0.2.0
|
14 |
+
transformers==4.49.0
|
15 |
+
huggingface_hub
|