Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -111,7 +111,7 @@ quantization_config = BitsAndBytesConfig(
|
|
111 |
load_in_4bit=True,
|
112 |
bnb_4bit_compute_dtype=torch.bfloat16
|
113 |
)
|
114 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
115 |
model = AutoModelForCausalLM.from_pretrained(
|
116 |
MODEL_ID,
|
117 |
device_map="auto",
|
|
|
111 |
load_in_4bit=True,
|
112 |
bnb_4bit_compute_dtype=torch.bfloat16
|
113 |
)
|
114 |
+
tokenizer = AutoTokenizer.from_pretrained('AIDC-AI/Marco-o1')
|
115 |
model = AutoModelForCausalLM.from_pretrained(
|
116 |
MODEL_ID,
|
117 |
device_map="auto",
|