Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -43,6 +43,8 @@ print(f"Starting to load the model to memory")
|
|
43 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
44 |
|
45 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
|
|
46 |
|
47 |
m = AutoModelForCausalLM.from_pretrained(
|
48 |
"google/gemma-2b-it",
|
|
|
43 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
44 |
|
45 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
46 |
+
prnit("HF_TOKEN")
|
47 |
+
|
48 |
|
49 |
m = AutoModelForCausalLM.from_pretrained(
|
50 |
"google/gemma-2b-it",
|