Hugging Face's logo

Spaces:
ginipick
/
time-machine
Running

App Files Community
ginipick commited on
Commit
a88da34
·
verified ·
1 Parent(s): 8ad6067

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ korean_labels = {
37
  }
38
 
39
  # load pipelines
40
- base_model = "black-forest-labs/FLUX.1-dev"
41
 
42
  taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16).to("cuda")
43
  pipe = FluxPipeline.from_pretrained(base_model,
 
37
  }
38
 
39
  # load pipelines
40
+ base_model = "black-forest-labs/FLUX.1-schnell"
41
 
42
  taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16).to("cuda")
43
  pipe = FluxPipeline.from_pretrained(base_model,