Spaces:
Sleeping
Sleeping
Sidharthan
commited on
Commit
·
a4e95d0
1
Parent(s):
97c8b2b
Fixing the adapter issues
Browse files
app.py
CHANGED
@@ -40,6 +40,7 @@ def load_model():
|
|
40 |
trust_remote_code=True,
|
41 |
low_cpu_mem_usage=True,
|
42 |
cache_dir='/app/cache'
|
|
|
43 |
).to(device)
|
44 |
|
45 |
return model, tokenizer
|
|
|
40 |
trust_remote_code=True,
|
41 |
low_cpu_mem_usage=True,
|
42 |
cache_dir='/app/cache'
|
43 |
+
|
44 |
).to(device)
|
45 |
|
46 |
return model, tokenizer
|