Hugging Face's logo

Spaces:
ginipick
/
Time-Stream
Paused

App Files Community
1
ginipick commited on
Commit
a5eb638
·
verified ·
1 Parent(s): b8e55c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -42,7 +42,10 @@ korean_labels = {
42
  }
43
 
44
  # load pipelines (토큰 인증 추가)
45
- base_model = "black-forest-labs/FLUX.1-dev"
 
 
 
46
 
47
  taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16, use_auth_token=hf_token).to("cuda")
48
  pipe = FluxPipeline.from_pretrained(base_model,
 
42
  }
43
 
44
  # load pipelines (토큰 인증 추가)
45
+ # base_model = "black-forest-labs/FLUX.1-dev"
46
+
47
+ base_model = "black-forest-labs/FLUX.1-schnell"
48
+
49
 
50
  taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16, use_auth_token=hf_token).to("cuda")
51
  pipe = FluxPipeline.from_pretrained(base_model,