Instructions to use not-pegasus/IMAGE_MODAL with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use not-pegasus/IMAGE_MODAL with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("not-pegasus/IMAGE_MODAL", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
| import functools | |
| import logging | |
| from typing import Optional, Tuple | |
| from pip._vendor.packaging import specifiers, version | |
| from pip._vendor.packaging.requirements import Requirement | |
| logger = logging.getLogger(__name__) | |
| def check_requires_python( | |
| requires_python: Optional[str], version_info: Tuple[int, ...] | |
| ) -> bool: | |
| """ | |
| Check if the given Python version matches a "Requires-Python" specifier. | |
| :param version_info: A 3-tuple of ints representing a Python | |
| major-minor-micro version to check (e.g. `sys.version_info[:3]`). | |
| :return: `True` if the given Python version satisfies the requirement. | |
| Otherwise, return `False`. | |
| :raises InvalidSpecifier: If `requires_python` has an invalid format. | |
| """ | |
| if requires_python is None: | |
| # The package provides no information | |
| return True | |
| requires_python_specifier = specifiers.SpecifierSet(requires_python) | |
| python_version = version.parse(".".join(map(str, version_info))) | |
| return python_version in requires_python_specifier | |
| def get_requirement(req_string: str) -> Requirement: | |
| """Construct a packaging.Requirement object with caching""" | |
| # Parsing requirement strings is expensive, and is also expected to happen | |
| # with a low diversity of different arguments (at least relative the number | |
| # constructed). This method adds a cache to requirement object creation to | |
| # minimize repeated parsing of the same string to construct equivalent | |
| # Requirement objects. | |
| return Requirement(req_string) | |