Update data_clean_final.py

data_clean_final.py

@@ -13,8 +13,7 @@ model = AutoModelForCausalLM.from_pretrained(
     model_name,
     device_map='auto',
     load_in_4bit=True,
-    torch_dtype='auto',
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+    torch_dtype='auto'
 )
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)