Spaces:
Paused
Paused
Ali Mohsin
commited on
Commit
·
1a4d50f
1
Parent(s):
4b464ab
eeqeqqe
Browse files- inference.py +2 -2
inference.py
CHANGED
|
@@ -46,7 +46,7 @@ class InferenceService:
|
|
| 46 |
for m in [self.resnet, self.vit]:
|
| 47 |
if m is not None:
|
| 48 |
for p in m.parameters():
|
| 49 |
-
|
| 50 |
|
| 51 |
# Update overall status
|
| 52 |
self.models_loaded = self.resnet_loaded and self.vit_loaded
|
|
@@ -178,7 +178,7 @@ class InferenceService:
|
|
| 178 |
for m in [self.resnet, self.vit]:
|
| 179 |
if m is not None:
|
| 180 |
for p in m.parameters():
|
| 181 |
-
|
| 182 |
|
| 183 |
# Update overall status
|
| 184 |
self.models_loaded = self.resnet_loaded and self.vit_loaded
|
|
|
|
| 46 |
for m in [self.resnet, self.vit]:
|
| 47 |
if m is not None:
|
| 48 |
for p in m.parameters():
|
| 49 |
+
p.requires_grad_(False)
|
| 50 |
|
| 51 |
# Update overall status
|
| 52 |
self.models_loaded = self.resnet_loaded and self.vit_loaded
|
|
|
|
| 178 |
for m in [self.resnet, self.vit]:
|
| 179 |
if m is not None:
|
| 180 |
for p in m.parameters():
|
| 181 |
+
p.requires_grad_(False)
|
| 182 |
|
| 183 |
# Update overall status
|
| 184 |
self.models_loaded = self.resnet_loaded and self.vit_loaded
|