Spaces:
Build error
Build error
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
|