Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
app.py
CHANGED
@@ -73,15 +73,15 @@ examples = [
|
|
73 |
print(f"### Cleaning cached examples ...")
|
74 |
os.system(f"rm -rf gradio_cached_examples/")
|
75 |
|
76 |
-
device =
|
77 |
|
78 |
class MagicTimeController:
|
79 |
def __init__(self):
|
80 |
# config dirs
|
81 |
self.basedir = os.getcwd()
|
82 |
-
self.stable_diffusion_dir = os.path.join(self.basedir,
|
83 |
-
self.motion_module_dir = os.path.join(self.basedir,
|
84 |
-
self.personalized_model_dir = os.path.join(self.basedir,
|
85 |
self.savedir = os.path.join(self.basedir, "outputs")
|
86 |
os.makedirs(self.savedir, exist_ok=True)
|
87 |
|
|
|
73 |
print(f"### Cleaning cached examples ...")
|
74 |
os.system(f"rm -rf gradio_cached_examples/")
|
75 |
|
76 |
+
device = "cuda"
|
77 |
|
78 |
class MagicTimeController:
|
79 |
def __init__(self):
|
80 |
# config dirs
|
81 |
self.basedir = os.getcwd()
|
82 |
+
self.stable_diffusion_dir = os.path.join(self.basedir, model_path, "Base_Model")
|
83 |
+
self.motion_module_dir = os.path.join(self.basedir, model_path, "Base_Model", "motion_module")
|
84 |
+
self.personalized_model_dir = os.path.join(self.basedir, model_path, "DreamBooth")
|
85 |
self.savedir = os.path.join(self.basedir, "outputs")
|
86 |
os.makedirs(self.savedir, exist_ok=True)
|
87 |
|