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
|