lifeng commited on
Commit
675ceb3
·
1 Parent(s): 10afe14

修改参数

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +7 -0
  3. requirements.txt +2 -1
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: EasyControl Ghibli
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