Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse filesfixed to always use float16
app.py
CHANGED
|
@@ -40,7 +40,7 @@ object_addition_finetuned_model_id = "paint-by-inpaint/add-finetuned-mb"
|
|
| 40 |
# general_editing_finetuned_model_id = "paint-by-inpaint/general-finetuned-mb"
|
| 41 |
|
| 42 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 43 |
-
dtype = torch.float16 if "cuda" in device else torch.float32
|
| 44 |
|
| 45 |
|
| 46 |
def load_model(model_id):
|
|
|
|
| 40 |
# general_editing_finetuned_model_id = "paint-by-inpaint/general-finetuned-mb"
|
| 41 |
|
| 42 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 43 |
+
dtype = torch.float16 #if "cuda" in device else torch.float32
|
| 44 |
|
| 45 |
|
| 46 |
def load_model(model_id):
|