Update app.py

app.py

@@ -20,7 +20,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel
 import torch.distributed
 if not hasattr(torch.distributed, 'tensor'):
-    torch.distributed.tensor = None  
+    class _FakeTensorModule:
+        class DTensor:
+            pass
+    torch.distributed.tensor = _FakeTensorModule()
 
 print("Torch:", torch.__version__)
 print("Transformers:", transformers.__version__)