Spaces:
Running
on
Zero
Running
on
Zero
fix display bug
Browse files
app.py
CHANGED
|
@@ -64,9 +64,8 @@ def load_model():
|
|
| 64 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
| 65 |
device_map="auto" if torch.cuda.is_available() else None,
|
| 66 |
trust_remote_code=True,
|
| 67 |
-
token=hf_token
|
| 68 |
-
|
| 69 |
-
)
|
| 70 |
|
| 71 |
print("β
Osmosis Structure model loaded successfully!")
|
| 72 |
return True
|
|
|
|
| 64 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
| 65 |
device_map="auto" if torch.cuda.is_available() else None,
|
| 66 |
trust_remote_code=True,
|
| 67 |
+
token=hf_token
|
| 68 |
+
)
|
|
|
|
| 69 |
|
| 70 |
print("β
Osmosis Structure model loaded successfully!")
|
| 71 |
return True
|