Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,8 +33,8 @@ except ImportError:
|
|
| 33 |
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
| 34 |
|
| 35 |
# Model configuration
|
| 36 |
-
MODEL_NAME = os.environ.get("MODEL_NAME", "/
|
| 37 |
-
TRANSFORMER_PATH = os.environ.get("TRANSFORMER_PATH", "/
|
| 38 |
|
| 39 |
dtype = torch.bfloat16
|
| 40 |
|
|
|
|
| 33 |
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
| 34 |
|
| 35 |
# Model configuration
|
| 36 |
+
MODEL_NAME = os.environ.get("MODEL_NAME", "Skywork/Unipic3-DMD")
|
| 37 |
+
TRANSFORMER_PATH = os.environ.get("TRANSFORMER_PATH", "Skywork/Unipic3-DMD/ema_transformer")
|
| 38 |
|
| 39 |
dtype = torch.bfloat16
|
| 40 |
|