zkmine commited on
Commit
d46e856
·
verified ·
1 Parent(s): 27cabc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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__)