Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
e5b7e3e
1
Parent(s):
14626ab
Fix indentation in load_infinity function for better readability
Browse files
app.py
CHANGED
@@ -220,10 +220,10 @@ def load_infinity(
|
|
220 |
state_dict = torch.load(model_path, map_location=device)
|
221 |
print(infinity_test.load_state_dict(state_dict))
|
222 |
|
223 |
-
|
224 |
-
|
225 |
|
226 |
-
|
227 |
|
228 |
def transform(pil_img, tgt_h, tgt_w):
|
229 |
width, height = pil_img.size
|
|
|
220 |
state_dict = torch.load(model_path, map_location=device)
|
221 |
print(infinity_test.load_state_dict(state_dict))
|
222 |
|
223 |
+
# Initialize random number generator on the correct device
|
224 |
+
infinity_test.rng = torch.Generator(device=device)
|
225 |
|
226 |
+
return infinity_test
|
227 |
|
228 |
def transform(pil_img, tgt_h, tgt_w):
|
229 |
width, height = pil_img.size
|