Python SDK for RockAI.online
Project description
Setup
Python SDK for inference and training for RockAI
# Install package using pip
pip install rockai-cli-app
from rockai_cli_app import Client
import asyncio
my_client = Client()
# Function to run the asynchronous function and print the result
async def main():
result = await my_client.run_async(
version="001bb81139b01780380407b4106ac681df46108e002eafbeb9ccb2d8faca42e1",
input={
"width": 1024,
"height": 1024,
"prompt": "a cartoon of a IRONMAN fighting with HULK, wall painting",
"guidance_scale": 3,
"negative_prompt": "(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
"num_inference_steps": 15,
}
)
print("Result:", result)
# Run the main function using asyncio.run
if __name__ == "__main__":
asyncio.run(main())
# clean docker space
docker system prune --all --force --volumes
Development
- Python version 3.11.5
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
rockai_cli_app-0.2.24.tar.gz
(15.5 kB
view hashes)
Built Distribution
Close
Hashes for rockai_cli_app-0.2.24-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b9b6264007edccda9fdcdc0a102a1ff3c430a8b9a5c6cd1e3ed395751b4a34dc |
|
MD5 | f7931d9b53f576813d2d1dd7fb21bcc6 |
|
BLAKE2b-256 | 2330ae48f3f70e807e2423d83843d6b42f2636ed174c40ce50052ba7a2357a27 |