Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2096,7 +2096,7 @@ loras = [
|
|
2096 |
#--------------------------------------------------Model Initialization-----------------------------------------------------------------------------------------#
|
2097 |
|
2098 |
dtype = torch.bfloat16
|
2099 |
-
device = "
|
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.#
|