Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Format image provider #66

Merged
merged 15 commits into from
May 1, 2023
Merged

Format image provider #66

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
29fc251
Adds image format
JasonWeill Apr 14, 2023
7ea615d
WIP: New provider for huggingface_image
JasonWeill Apr 14, 2023
61fafc5
Adds image provider to __init__
JasonWeill Apr 14, 2023
34e2977
Improve magics.
Apr 15, 2023
e5fb010
Fixing metadata handling.
ellisonbg Apr 16, 2023
68fd90e
Updates user docs
JasonWeill Apr 17, 2023
41bab6e
WIP: Prints debug output on initial load
JasonWeill Apr 18, 2023
876b1f5
Removes debug info from load
JasonWeill Apr 25, 2023
bffe4fe
Removes separate provider for images
JasonWeill Apr 25, 2023
f03dfe8
Outputs image in Base64 format
JasonWeill Apr 25, 2023
2f097f2
Removes HF image provider, updates docs
JasonWeill Apr 25, 2023
8befd7c
Removes limitation of only supporting JPEG format
JasonWeill Apr 26, 2023
754072d
Promotes VALID_TASKS out of class
JasonWeill Apr 26, 2023
3064af1
Updates error handling, image generation example notebook
JasonWeill Apr 27, 2023
0211aef
Updates dependencies per @dlqqq
JasonWeill Apr 28, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Promotes VALID_TASKS out of class
  • Loading branch information
@JasonWeill
JasonWeill committed Apr 28, 2023
commit 754072d09d2cf2ff026cbaa950556b213e7d9f5c
10 changes: 5 additions & 5 deletions packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,8 @@ class CohereProvider(BaseProvider, Cohere):
pypi_package_deps = ["cohere"]
auth_strategy = EnvAuthStrategy(name="COHERE_API_KEY")

HUGGINGFACE_HUB_VALID_TASKS = ("text2text-generation", "text-generation", "text-to-image")

class HfHubProvider(BaseProvider, HuggingFaceHub):
id = "huggingface_hub"
name = "HuggingFace Hub"
Expand All @@ -152,18 +154,16 @@ def validate_environment(cls, values: Dict) -> Dict:
try:
from huggingface_hub.inference_api import InferenceApi

VALID_TASKS = ("text2text-generation", "text-generation", "text-to-image")

repo_id = values["repo_id"]
client = InferenceApi(
repo_id=repo_id,
token=huggingfacehub_api_token,
task=values.get("task"),
)
if client.task not in VALID_TASKS:
if client.task not in HUGGINGFACE_HUB_VALID_TASKS:
raise ValueError(
f"Got invalid task {client.task}, "
f"currently only {VALID_TASKS} are supported"
f"currently only {HUGGINGFACE_HUB_VALID_TASKS} are supported"
)
values["client"] = client
except ImportError:
Expand Down Expand Up @@ -219,7 +219,7 @@ def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
else:
raise ValueError(
f"Got invalid task {self.client.task}, "
f"currently only {VALID_TASKS} are supported"
f"currently only {HUGGINGFACE_HUB_VALID_TASKS} are supported"
)
if stop is not None:
# This is a bit hacky, but I can't figure out a better way to enforce
Expand Down