MisterAI commited on
Commit
a6bd83c
·
verified ·
1 Parent(s): 8f63867

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ class FluxGenerator:
53
  # Configuration pour minimiser l'utilisation de la mémoire
54
  model_kwargs = {
55
  "low_cpu_mem_usage": True,
56
- "torch_dtype": torch.float16, # Utilise float16 pour réduire la mémoire
57
  "use_safetensors": True, # Utilise safetensors pour un chargement plus efficace
58
  }
59
 
 
53
  # Configuration pour minimiser l'utilisation de la mémoire
54
  model_kwargs = {
55
  "low_cpu_mem_usage": True,
56
+ "torch_dtype": torch.float8, # Utilise float16 pour réduire la mémoire
57
  "use_safetensors": True, # Utilise safetensors pour un chargement plus efficace
58
  }
59