Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
|
|
20 |
model_name = "meta-llama/Meta-Llama-3.1-8B"
|
21 |
try:
|
22 |
model_llama = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.float16,
|
23 |
-
load_in_8bit=True
|
24 |
except OSError as e:
|
25 |
print(f"Error loading model: {e}")
|
26 |
|
|
|
20 |
model_name = "meta-llama/Meta-Llama-3.1-8B"
|
21 |
try:
|
22 |
model_llama = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.float16,
|
23 |
+
load_in_8bit=True, token = hf_token) # use device_map for automatic sharding
|
24 |
except OSError as e:
|
25 |
print(f"Error loading model: {e}")
|
26 |
|