Skip to content

Instantly share code, notes, and snippets.

View deezeddd's full-sized avatar
😃
Job Hunt | Open-Source

Vedant deezeddd

😃
Job Hunt | Open-Source
View GitHub Profile
import time
import os
import logging
import random
from datasets import load_dataset
class QuantAutoGPTQ:
def __init__(self, model_name_or_path, output_dir, dataset,
num_samples=128, trust_remote_code=False, cache_examples=True,
use_fast=True, use_triton=False, bits=[4], group_size=[128], damp=[0.01],