Spaces:
Runtime error
Runtime error
João Artur
commited on
Commit
·
321391f
1
Parent(s):
e227425
cache
Browse files
.env
ADDED
|
File without changes
|
app.py
CHANGED
|
@@ -1,5 +1,9 @@
|
|
| 1 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 2 |
import torch
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
|
| 4 |
# Load LLaMA 3 model and tokenizer
|
| 5 |
model_name = "path_to_llama3_model" # Replace with LLaMA 3 model path or Hugging Face model link if available
|
|
|
|
| 1 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 2 |
import torch
|
| 3 |
+
import os
|
| 4 |
+
|
| 5 |
+
# Set a writable cache directory for Hugging Face
|
| 6 |
+
os.environ['TRANSFORMERS_CACHE'] = './cache'
|
| 7 |
|
| 8 |
# Load LLaMA 3 model and tokenizer
|
| 9 |
model_name = "path_to_llama3_model" # Replace with LLaMA 3 model path or Hugging Face model link if available
|