Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
|
@@ -6,6 +6,9 @@ import torch
|
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
import numpy as np
|
| 8 |
import io
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
# Load the SDXL base model pipeline
|
| 11 |
base = DiffusionPipeline.from_pretrained(
|
|
|
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
import numpy as np
|
| 8 |
import io
|
| 9 |
+
from transformers.utils import move_cache
|
| 10 |
+
move_cache()
|
| 11 |
+
|
| 12 |
|
| 13 |
# Load the SDXL base model pipeline
|
| 14 |
base = DiffusionPipeline.from_pretrained(
|