Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
c75a3f7
1
Parent(s):
0b18eec
update
Browse files- test_code/inference.py +2 -0
test_code/inference.py
CHANGED
|
@@ -336,6 +336,8 @@ def execute_inference(huggingface_pretrained_path, model_type, validation_path,
|
|
| 336 |
)
|
| 337 |
|
| 338 |
|
|
|
|
|
|
|
| 339 |
# For text ..............................................
|
| 340 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 341 |
base_config["pretrained_tokenizer_name_or_path"],
|
|
|
|
| 336 |
)
|
| 337 |
|
| 338 |
|
| 339 |
+
print("device we have is ", accelerator.device)
|
| 340 |
+
|
| 341 |
# For text ..............................................
|
| 342 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 343 |
base_config["pretrained_tokenizer_name_or_path"],
|