[Tests] Fix test case; update structure (#1375)

# Summary
- Fix device_map and set torch.dtype for the given model
- Move tests to a folder which makes more sense

Author: dsikka

tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune.py

@@ -64,8 +64,7 @@ def _test_oneshot_and_finetune(self):
         )
         # model is first sparsified, then finetuned, both should have the same sparsity
         assert config_sparse_applied["global_sparsity"] == pytest.approx(
-            config_finetune_applied["global_sparsity"],
-            abs=1e-5
+            config_finetune_applied["global_sparsity"], abs=1e-5
         )
 
     def tearDown(self):

tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune_with_tokenizer.py

@@ -36,7 +36,7 @@ def test_oneshot_and_finetune_with_tokenizer(self):
             self.model,
         )
         model_loaded = AutoModelForCausalLM.from_pretrained(
-            self.model, device_map="auto"
+            self.model, device_map="cuda:0", torch_dtype="auto"
         )
 
         dataset_loaded = load_dataset(

tests/llmcompressor/transformers/obcq/obcq_configs/sparsity_generic/config.yaml

@@ -0,0 +1,4 @@
+cadence: "nightly"
+test_type: "regression"
+model: "nm-testing/llama2.c-stories15M"
+dataset: open_platypus

…e/test_finetune_oneshot_with_modifier.py …rmers/obcq/test_oneshot_with_modifier.pytests/llmcompressor/transformers/finetune/test_finetune_oneshot_with_modifier.py renamed to tests/llmcompressor/transformers/obcq/test_oneshot_with_modifier.py tests/llmcompressor/transformers/finetune/test_finetune_oneshot_with_modifier.py renamed to tests/llmcompressor/transformers/obcq/test_oneshot_with_modifier.py

@@ -8,7 +8,9 @@
 
 from tests.testing_utils import parse_params, requires_gpu
 
-CONFIGS_DIRECTORY = "tests/llmcompressor/transformers/finetune/finetune_generic"
+CONFIGS_DIRECTORY = (
+    "tests/llmcompressor/transformers/obcq/obcq_configs/sparsity_generic"
+)
 
 
 @pytest.mark.integration