Arashpey commited on
Commit
3127aff
·
verified ·
1 Parent(s): d174d61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2096,7 +2096,7 @@ loras = [
2096
  #--------------------------------------------------Model Initialization-----------------------------------------------------------------------------------------#
2097
 
2098
  dtype = torch.bfloat16
2099
- device = "cuda" if torch.cuda.is_available() else "cpu"
2100
  base_model = "black-forest-labs/FLUX.1-dev"
2101
 
2102
  #TAEF1 is very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE. FLUX.1 is useful for real-time previewing of the FLUX.1 generation process.#
 
2096
  #--------------------------------------------------Model Initialization-----------------------------------------------------------------------------------------#
2097
 
2098
  dtype = torch.bfloat16
2099
+ device = "cpu"
2100
  base_model = "black-forest-labs/FLUX.1-dev"
2101
 
2102
  #TAEF1 is very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE. FLUX.1 is useful for real-time previewing of the FLUX.1 generation process.#