saranbalan commited on
Commit
56d0e1d
·
verified ·
1 Parent(s): 656ef1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -8,6 +8,11 @@ from diffusers import StableDiffusionPipeline
8
  import matplotlib.pyplot as plt
9
  import torch
10
  from huggingface_hub import InferenceApi
 
 
 
 
 
11
 
12
  # Set up Groq API key
13
  api_key = os.getenv("GROQ_API_KEY")
 
8
  import matplotlib.pyplot as plt
9
  import torch
10
  from huggingface_hub import InferenceApi
11
+ from transformers.utils import move_cache
12
+
13
+ # Migrate the cache manually
14
+ move_cache()
15
+
16
 
17
  # Set up Groq API key
18
  api_key = os.getenv("GROQ_API_KEY")