Skip to content

Commit f756d80

Browse files
bigcat88marawan206
andauthored
[Partner Nodes] Topaz Astra 2 model (Comfy-Org#13672)
* feat(api-nodes): add Topaz Astra 2 model Signed-off-by: bigcat88 <bigcat88@icloud.com> * feat(api-nodes): make Astra 2 the default Topaz upscaler model Reorder UPSCALER_MODELS_MAP and the upscaler_model dynamic combo so "Astra 2" appears first, surfacing it as the default selection. --------- Signed-off-by: bigcat88 <bigcat88@icloud.com> Co-authored-by: Marwan Mostafa <marawan206@gmail.com>
1 parent 1d23a87 commit f756d80

2 files changed

Lines changed: 365 additions & 5 deletions

File tree

comfy_api_nodes/apis/topaz.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Optional, Union
1+
from typing import Optional
22

33
from pydantic import BaseModel, Field
44

@@ -72,8 +72,11 @@ class VideoEnhancementFilter(BaseModel):
7272
grain: Optional[float] = Field(None, description="Grain after AI model processing")
7373
grainSize: Optional[float] = Field(None, description="Size of generated grain")
7474
recoverOriginalDetailValue: Optional[float] = Field(None, description="Source details into the output video")
75-
creativity: Optional[str] = Field(None, description="Creativity level(high, low) for slc-1 only")
75+
creativity: float | str | None = Field(None, description="slc-1/slp-2.5: enum (low/middle/high). ast-2: decimal 0.0-1.0.")
7676
isOptimizedMode: Optional[bool] = Field(None, description="Set to true for Starlight Creative (slc-1) only")
77+
prompt: str | None = Field(None, description="Descriptive scene prompt (ast-2 only)")
78+
sharp: float | None = Field(None, description="ast-2 pre-enhance sharpness")
79+
realism: float | None = Field(None, description="ast-2 realism control")
7780

7881

7982
class OutputInformationVideo(BaseModel):
@@ -90,7 +93,7 @@ class Overrides(BaseModel):
9093

9194
class CreateVideoRequest(BaseModel):
9295
source: CreateVideoRequestSource = Field(...)
93-
filters: list[Union[VideoFrameInterpolationFilter, VideoEnhancementFilter]] = Field(...)
96+
filters: list[VideoFrameInterpolationFilter | VideoEnhancementFilter] = Field(...)
9497
output: OutputInformationVideo = Field(...)
9598
overrides: Overrides = Field(Overrides(isPaidDiffusion=True))
9699

0 commit comments

Comments
 (0)