Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,6 +18,9 @@ import transformers
|
|
| 18 |
import peft
|
| 19 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 20 |
from peft import PeftModel
|
|
|
|
|
|
|
|
|
|
| 21 |
|
| 22 |
print("Torch:", torch.__version__)
|
| 23 |
print("Transformers:", transformers.__version__)
|
|
|
|
| 18 |
import peft
|
| 19 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 20 |
from peft import PeftModel
|
| 21 |
+
import torch.distributed
|
| 22 |
+
if not hasattr(torch.distributed, 'tensor'):
|
| 23 |
+
torch.distributed.tensor = None
|
| 24 |
|
| 25 |
print("Torch:", torch.__version__)
|
| 26 |
print("Transformers:", transformers.__version__)
|