Update README.md

README.md

@@ -61,9 +61,9 @@ You can run the smashed model with these steps:
    from transformers import AutoModelForCausalLM, AutoTokenizer
 
     try: 
-    model = HQQModelForCausalLM.from_quantized("PrunaAI/facebook-opt-125m-HQQ-1bit-smashed", device_map='auto')
-except: 
-    model = AutoHQQHFModel.from_quantized("PrunaAI/facebook-opt-125m-HQQ-1bit-smashed")
+      model = HQQModelForCausalLM.from_quantized("PrunaAI/facebook-opt-125m-HQQ-1bit-smashed", device_map='auto')
+    except: 
+      model = AutoHQQHFModel.from_quantized("PrunaAI/facebook-opt-125m-HQQ-1bit-smashed")
     tokenizer = AutoTokenizer.from_pretrained("facebook/opt-125m")
     
     input_ids = tokenizer("What is the color of prunes?,", return_tensors='pt').to(model.device)["input_ids"]