Spaces:
Running
Running
Commit
·
2ad01cf
1
Parent(s):
fb1d381
update
Browse files- gradio_demo/app.py +2 -0
gradio_demo/app.py
CHANGED
|
@@ -96,6 +96,8 @@ aggregator_state_dict = torch.load(
|
|
| 96 |
pipe.aggregator.load_state_dict(aggregator_state_dict, strict=True)
|
| 97 |
pipe.aggregator.to(device=device, dtype=torch_dtype)
|
| 98 |
|
|
|
|
|
|
|
| 99 |
MAX_SEED = np.iinfo(np.int32).max
|
| 100 |
MAX_IMAGE_SIZE = 1024
|
| 101 |
|
|
|
|
| 96 |
pipe.aggregator.load_state_dict(aggregator_state_dict, strict=True)
|
| 97 |
pipe.aggregator.to(device=device, dtype=torch_dtype)
|
| 98 |
|
| 99 |
+
print("******loaded")
|
| 100 |
+
|
| 101 |
MAX_SEED = np.iinfo(np.int32).max
|
| 102 |
MAX_IMAGE_SIZE = 1024
|
| 103 |
|