fix error
Browse files- app.py +3 -0
- 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
|