Spaces:
Runtime error
Runtime error
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 |
|