Spaces:
Runtime error
Runtime error
Update merged_files3.py
Browse files- merged_files3.py +2 -1
merged_files3.py
CHANGED
|
@@ -81,7 +81,8 @@ if torch.cuda.is_available():
|
|
| 81 |
device = torch.device('cuda')
|
| 82 |
else:
|
| 83 |
device = torch.device('cpu')
|
| 84 |
-
|
|
|
|
| 85 |
|
| 86 |
MAX_IMAGE_WIDTH = 2048
|
| 87 |
IMAGE_FORMAT = "JPEG"
|
|
|
|
| 81 |
device = torch.device('cuda')
|
| 82 |
else:
|
| 83 |
device = torch.device('cpu')
|
| 84 |
+
|
| 85 |
+
dtype = torch.float16 # Use float16 consistently for all models
|
| 86 |
|
| 87 |
MAX_IMAGE_WIDTH = 2048
|
| 88 |
IMAGE_FORMAT = "JPEG"
|