yasserrmd commited on
Commit
d2d56e8
1 Parent(s): b5b4791

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,6 +3,9 @@ from diffusers import FluxPipeline
3
  from transformers import pipeline
4
  import gradio as gr
5
  import spaces
 
 
 
6
 
7
 
8
  device=torch.device('cuda')
 
3
  from transformers import pipeline
4
  import gradio as gr
5
  import spaces
6
+ from transformers.util.hub import mvoe_cache
7
+
8
+ move_cache()
9
 
10
 
11
  device=torch.device('cuda')