Spaces:

saakshigupta
/

deepfake-explainer-app

Paused

saakshigupta commited on Apr 2

Commit

cca832d

verified ·

1 Parent(s): b9a4f3a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,6 @@ def load_model():
         model = AutoModelForCausalLM.from_pretrained(
             base_model_id,
             device_map="auto",
-            load_in_4bit=True,
             quantization_config=BitsAndBytesConfig(
                 load_in_4bit=True,
                 bnb_4bit_compute_dtype=torch.float16,

         model = AutoModelForCausalLM.from_pretrained(
             base_model_id,
             device_map="auto",
             quantization_config=BitsAndBytesConfig(
                 load_in_4bit=True,
                 bnb_4bit_compute_dtype=torch.float16,