Spaces:
Running
on
Zero
Running
on
Zero
brandonsmart
commited on
Commit
•
e059ab1
1
Parent(s):
7dfcb4e
Attempt to resolve CUDA issue in spaces
Browse files
demo.py
CHANGED
@@ -55,7 +55,8 @@ if __name__ == '__main__':
|
|
55 |
model_name = "brandonsmart/splatt3r_v1.0"
|
56 |
filename = "epoch=19-step=1200.ckpt"
|
57 |
weights_path = hf_hub_download(repo_id=model_name, filename=filename)
|
58 |
-
model = main.MAST3RGaussians.load_from_checkpoint(weights_path,
|
|
|
59 |
chkpt_tag = hash_md5(weights_path)
|
60 |
|
61 |
# Define example inputs and their corresponding precalculated outputs
|
|
|
55 |
model_name = "brandonsmart/splatt3r_v1.0"
|
56 |
filename = "epoch=19-step=1200.ckpt"
|
57 |
weights_path = hf_hub_download(repo_id=model_name, filename=filename)
|
58 |
+
model = main.MAST3RGaussians.load_from_checkpoint(weights_path, 'cpu')
|
59 |
+
model = model.to(device)
|
60 |
chkpt_tag = hash_md5(weights_path)
|
61 |
|
62 |
# Define example inputs and their corresponding precalculated outputs
|