huanngzh commited on
Commit
aef4e02
1 Parent(s): 1f7edc1
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -8,6 +8,9 @@ import torch
8
 
9
  from inference_t2mv_sdxl import prepare_pipeline, run_pipeline
10
 
 
 
 
11
 
12
  # Base model
13
  base_model = "stabilityai/stable-diffusion-xl-base-1.0"
 
8
 
9
  from inference_t2mv_sdxl import prepare_pipeline, run_pipeline
10
 
11
+ import transformers
12
+
13
+ transformers.utils.move_cache()
14
 
15
  # Base model
16
  base_model = "stabilityai/stable-diffusion-xl-base-1.0"
requirements.txt CHANGED
@@ -12,4 +12,4 @@ omegaconf
12
  trimesh
13
  einops
14
  gradio==4.44.1
15
- numpy
 
12
  trimesh
13
  einops
14
  gradio==4.44.1
15
+ numpy==1.26.2