Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ 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 |
|
48 |
|
49 |
m = AutoModelForCausalLM.from_pretrained(
|
|
|
43 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
44 |
|
45 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
46 |
+
print("HF_TOKEN")
|
47 |
|
48 |
|
49 |
m = AutoModelForCausalLM.from_pretrained(
|